LintCode407

来源:互联网 发布:冷烟花在淘宝上怎么买 编辑:程序博客网 时间:2024/06/04 19:47

加一

给定一个非负数,表示一个数字数组,在该数的基础上+1,返回一个新的数组。

该数字按照大小进行排列,最大的数在列表的最前面。

样例

给定 [1,2,3] 表示 123, 返回 [1,2,4].

给定 [9,9,9] 表示 999, 返回 [1,0,0,0].

代码://老套的逢10近1。注意各种条件细节。。。

class Solution {
public:
    /**
     * @param digits a number represented as an array of digits
     * @return the result
     */
    vector<int> plusOne(vector<int>& digits) {
        // Write your code here
        digits[digits.size()-1]+=1;
        if(digits.size()==1)
        {
            if(digits[0]==10)
            {
                digits.push_back(0);
                digits[0]=1;
            }
        } else {
            for(int i=digits.size()-1;i>=1;i--)
                if(digits[i]==10){
                    digits[i]=0;
                    digits[i-1]+=1;
                }
        }
        if(digits[0]==10){
            digits.push_back(digits[digits.size()-1]);
            for(int i=digits.size()-3;i>=1;i--)
                digits[i+1]=digits[i];
            digits[0]=1;
        }
        return digits;
    }
};

原创粉丝点击