【leetcode】66. 加一(Plus One)

---

题目描述

【leetcode】66. 加一(Plus One)
给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。

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

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

第一次解答

思路:
从后往前加，后一个数+1结果若>=10
则前一个数+1
该方法缺点：对于[9,9,9,9,9]这种数，遍历到最后一个9时要进一位，
整体需要往后移动一位，存在内存拷贝开销

test case:
[1,2,3]
[4,3,2,1]
[9,9,9,9]
[0]

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int index = digits.size()-1;
        digits[index] += 1;
        while(digits[index] >= 10){
            digits[index] = 0;
            --index;
            if(index < 0){
                //插入元素
                digits.insert(digits.begin(), 1);
                return digits;
            }
            else{
                digits[index] += 1;
            }
        }
        return digits;
    }
};

结果：

第二次解答

看了这个题解，思路几乎不变，感觉代码还可以优化下，可以更简洁，理论上可以有些许提高。

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

结果：

相关/参考链接