66.加一
题目描述:
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
题目思路:
本题相当于一个加法运算,就是每次加数都为1,加法从最后一位加起,如果最后一位为9则需要进1,设置进位flag,因为只加1,进位也最多加1,如果最后一位为9,那么flag=1,代表需要继续进位,并将 当前位 置为0;此时继续循环,到达前一位,如果不是9,代表不需要进位,此时将当前位加1后直接输出。
如果是9999,那么每一位都需要进位,按照上述方法,循环结束后,i=-1,flag=1并且此时数组值为0000,那么就将第一位置为1,并在最后一位加上一个0;
代码:
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
int len = digits.size();
int flag=1;//flag代表进位
int i;
for(i=len-1;i>=0&&flag==1;i--)
{
if(digits[i]==9)
{
digits[i]=0;
flag=1;
}
else
{
digits[i]=digits[i]+flag;
flag=0;
}
}
if(i==-1&&flag==1)
{
digits[i+1]=1;
digits.push_back(0);//在数组尾部加值
}
return digits;
}
};