[LeetCode] 66. 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.

题意是给一个非空数组，表示一个数字，请对其加一。思路是从右往左扫，判断当前位置上的数字digits[i]是不是9，两种情况

• 第一种情况，如果digits[i]是9，就把当前位digits[i]变为0
• 第二种情况，就正常对当前位+1，立即退出循环，返回数组

若是第一种情况，就说明for循环一直跑到了最后，所以需要在最后开一个新数组，长度为原数组长度+1。如果是第二种情况，for循环是跑不到最后的，因为程序在做完+1的动作之后立马退出了。同时注意，当遇到9的时候，因为还是在for循环中，所以即使当前位被置为0了，他的下一位还是会被+1。

时间O(n)

空间O(n) - res数组

Java实现

class Solution {
    public int[] plusOne(int[] digits) {
        // corner case
        if (digits == null || digits.length == 0) {
            return digits;
        }

        // normal case
        for (int i = digits.length - 1; i >= 0; i--) {
            if (digits[i] < 9) {
                digits[i]++;
                return digits;
            } else {
                digits[i] = 0;
            }
        }
        int[] res = new int[digits.length + 1];
        res[0] = 1;
        return res;
    }
}

JavaScript实现

/**
 * @param {number[]} digits
 * @return {number[]}
 */
var plusOne = function(digits) {
    // corner case
    if (digits == null || digits.length == 0) {
        return digits;
    }

    // normal case
    for (let i = digits.length - 1; i >= 0; i--) {
        if (digits[i] < 9) {
            digits[i]++;
            return digits;
        } else {
            digits[i] = 0;
        }
    }
    let res = new Array(digits.length + 1).fill(0);
    res[0] = 1;
    return res;
};

LeetCode 题目总结