题目
给定一个由
整数
组成的
非空
数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储
单个
数字。
你可以假设除了整数 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]
代码
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
int n = digits.size();
for(int i = n-1; i >= 0; i--){
if(digits[i]!=9){
digits[i]++;
for(int j = i+1;j<n;j++) digits[j]=0;
return digits;
}
}
vector<int> ans(n+1);
ans[0]=1;
return ans;
}
};
解析
分情况讨论
1.末尾没有9的,直接++
2.末尾有9,但是不全是9的,从后向前找到不是9的++,后面的直接赋0
3.全都是9的,直接构造一个n+1长度的数组,第一个为1后面都是0
又是一些小细节
1.虽然说是分情况讨论,但大可不必一上来就if,这样超复杂,最后你会迷失在大括号的层级里(别问我是怎么知道的)
2.解析里的第三点,没有想到可以直接构造新的数组,自己做的麻烦了很多
3.解析中1 2点可以合并,如果末尾没有9直接不走for嘛,是没问题的。
4.总的来说没想到可以这么简单,自己写的又臭又长而且还会卡测试用例
5.c++构造数组: vector < int > ans(n)