给定一个非负整数组成的非空数组,在该数的基础上加一,返回一个新的数组。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
思路:就是用数组表示非负整数,将这个非负整数加1,用数组返回。
大致分了三种情况:
1、从个位起,没有9的话,直接在尾数加1.例如1234,123.
2、有9但不全为9,例如129,1239,把9那位变0,下一位+1.
3、所有元素都是9,例如9,999这样的,就新建一个数组,长度为原数组长度+1,第一位置为1就好.例如999->10000.
代码如下:
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
for(int i=digits.size()-1;i>=0;i--) //从最后一个元素向前遍历
if(digits[i]==9) //如果是9就置为0
digits[i]=0;
else
{
digits[i]++; //不是9就++,数组加一完成,返回更改后的原数组
return digits;
}
vector<int>digits2(digits.size()+1); //全为9的情况,就新建一个数组
digits2[0]=1;
return digits2;
}
};