66. 加一
题目描述
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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 ≤ d i g i t s . l e n g t h ≤ 100 1 \le digits.length \le 100 1≤digits.length≤100
- 0 ≤ d i g i t s [ i ] ≤ 9 0 \le digits[i] \le 9 0≤digits[i]≤9
题解:
模拟竖式加法。
注意,此题将数组逆序比较容易操作,因为 99999....
加一会产生进位,把高位放在尾部方便添加新的元素。
时间复杂度: O ( n ) O(n) O(n)
额外空间复杂度: O ( 1 ) O(1) 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%
*/