力扣-加一

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

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

你可以假设除了整数 0 之外,这个整数不会以零开头。
力扣-加一_第1张图片
解题思路:
首先,当最后一位元素不是9,直接加一返回即可;当最后一位元素是9时,考虑逐级判断进位。

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

你可能感兴趣的:(力扣,刷题,leetcode)