【LeetCode】加一

给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。

最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。

你可以假设除了整数 0 之外,这个整数不会以零开头。

示例 1:

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。

示例 2:

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

我对这道题的思路很简单,就是对最后一位加一然后输出,对于一种需要进位的情况,就需要用到vector这个STL的用法,那就是用push_back来在最后一位加0,并且把除了第一位的所有位变为0,然后将第一位置1,但是空间消耗较多,因为是逐一遍历的,下面给出AC代码:

class Solution {
public:
    vector plusOne(vector& digits) {
        int len=digits.size();
        if(len==1&&digits[0]!=9){
            digits[0]+=1;
            return digits;
        }
        if(len==1&&digits[0]==9){
            digits[0]=1;
            digits.push_back(0);
            return digits;
        }
        else{
        digits[len-1]+=1;
        for(int i=len-1;i>=0;i--){
            if(i==0)
                break;
            else{
                if(digits[i]==10&&i!=0){
                    digits[i]=digits[i]%10;
                    digits[i-1]+=1;
                    if(digits[i-1]==10&&i-1==0){
                        for(int i=len-1;i>0;i--){
                            digits[i]=0;   
                            }
                        digits[0]=1;
                        digits.push_back(0);
                    }
                }
            }
        }
    }    
        return digits;
    }
};

 

你可能感兴趣的:(CCF备考)