【LeetCode】数组初级算法-加一

题目描述

LeetCode：
https://leetcode-cn.com/explo...

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

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

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

示例 1:

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

示例 2:

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

思路

注意处理进位，用flag标识是否进位。

JavaScript实现

/**
 * @param {number[]} digits
 * @return {number[]}
 */
var plusOne = function(digits) {
    let flag = false;
    for(let i = digits.length-1; i >= 0; i-- ){
        if(digits[i] + 1 < 10){
            digits[i]++;
            flag = false;
            break;
        }else{
            digits[i] = 0;
            flag = true;
        }
    }
    if(flag){
        digits[0] = 1;
        digits.push(0);
    }
    return digits;
};