题目描述
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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的数字存在两种情况
- 除9之外的数字(直接返回加1的值)
- 数字9(需要考虑是否要进位)
class Solution {//cpp
public:
vector<int> plusOne(vector<int>& digits) {
for(int i = digits.size() - 1; i >= 0; i--){
if(digits[i] != 9){
digits[i]++;
return digits;
}
digits[i] = 0;
}
// 若跳出for循环,说明数组元素全为9
digits.insert(digits.begin(), 1);//数组digits中迭代器指向元素(数组头)前增加一个元素1
return digits;
/*
vector<int> res(digits.size() + 1);
res[0] = 1;
for(int i = 1; i < res.size(); i++){
res[i] = 0;
}
return res;
*/
}
};
class Solution:#python
def plusOne(self, digits: List[int]) -> List[int]:
carry = 1
for i in range(len(digits) - 1, -1, -1):
digits[i], carry = (carry + digits[i]) % 10, (carry + digits[i]) // 10 #求余, 向下整除
#将第i位数字加1再更新,如果第i位加1为10,则需要进位,将carry置为1,这样再往前遍历时前1位数字加1,如果不需要进位,carry置为0,往前遍历时前1位数字不变
return [carry] + digits if carry else digits
#如果carry为1,说明最高位为9(原数组所有元素为9)且加1变为0,需要在最前面增加一位1