算法 - Plus one

Given a non-empty array of digits representing a non-negative integer, plus one to the integer.

The digits are stored such that the most significant digit is at the head of the list, and each element in the array contain a single digit.

You may assume the integer does not contain any leading zero, except the number 0 itself.

Example 1:

Input: [1,2,3]
Output: [1,2,4]
Explanation: The array represents the integer 123.

Example 2:

Input: [4,3,2,1]
Output: [4,3,2,2]
Explanation: The array represents the integer 4321.

思路

本题主要考虑进位的问题，最后一位为9，需要向前进位；如果所有位为9，需要新增一位放在第一位。

解法

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

 

拓展：如果这个序列用的是链表，该如何处理？

思路

因为链表只能单向遍历，所有有递归（用栈和不用栈的2中解法），下面考虑一种巧妙的处理：

• 遍历链表，找到右起第一个不为9的数字，如果找不到这样的数字，说明所有数字均为9，那么在表头新建一个值为0的新节点，进行加1处理，然后把右边所有的数字都置为0即可