leetcode-66:加一

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 plusOne(vector& 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;
    }
};

你可能感兴趣的:(C++,leetcode,leetcode)