欢迎访问LeetCode题解合集
题目描述
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入:digits = [1,2,3]
输出:[1,2,4]
解释:输入数组表示数字 123。
示例 2:
输入:digits = [4,3,2,1]
输出:[4,3,2,2]
解释:输入数组表示数字 4321。
示例 3:
输入:digits = [0]
输出:[1]
提示:
- $1 \le digits.length \le 100$
- $0 \le digits[i] \le 9$
题解:
模拟竖式加法。
注意,此题将数组逆序比较容易操作,因为 99999....
加一会产生进位,把高位放在尾部方便添加新的元素。
时间复杂度:$O(n)$
额外空间复杂度:$O(1)$
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
reverse(digits.begin(), digits.end());
digits[0] += 1;
int d = 0;
for ( int i = 0; i < digits.size(); ++i ) {
d = digits[i] / 10;
if ( !d ) break;
if ( d ) {
digits[i] %= 10;
if ( i + 1 < digits.size() ) digits[i + 1] += d;
else break;
}
}
if ( d ) digits.push_back( d );
reverse(digits.begin(), digits.end());
return digits;
}
};
/*
时间:0ms,击败:100.00%
内存:8.4MB,击败:99.56%
*/