Plus One

最新推荐文章于 2021-07-27 15:03:42 发布

原创最新推荐文章于 2021-07-27 15:03:42 发布 · 810 阅读

0 ·

CC 4.0 BY-SA版权

数组/集合 (array / set) 专栏收录该内容

28 篇文章

订阅专栏

本文介绍了一种将数组表示的数字加一的算法，并通过Java代码实现。该算法首先判断数组是否全为9，如果是，则在最前面添加1并返回新数组；否则，从数组末尾开始遍历，找到第一个不为9的元素加1，并处理进位。

Given a number represented as an array of digits, plus one to the number.

public class Solution {
    public int[] plusOne(int[] digits) {
        if (digits == null || digits.length == 0) return null;
        boolean allNine = true;
        for (int i = 0; i < digits.length; i++) {
            if (digits[i] != 9) {
                allNine = false;
                break;
            }
        }
        int[] result;
        if (allNine == true) {
            result = new int[digits.length + 1];
            result[0] = 1;
            return result;
        } else {
            result = new int[digits.length];
        }
        
        int value = 0;
        int extra = 0;
        for (int i = digits.length - 1; i >= 0; i--) {
            if (i == digits.length - 1) {
                value = digits[i] + extra + 1; 
            } else {
                value = digits[i] + extra;
            }            
            extra = value / 10;
            value = value % 10;
            result[i] = value;
        }
        return result;
    }
}