LeetCode每日一题: 加一(No.66)

378 阅读1分钟

题目:加一


给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。

示例:


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

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

思考:


将原数组从后向前遍历,若最后一个元素即个位数字小于9,直接+1就行。
若大于9,则要向前进位,当前位直接赋值为0,下次循环将倒数第二位+1。
从后向前以此类推,遇到小于9的直就接+1返回结果。
否则直到循环结束则说明原数是“99”、“999”类似这样的,此时加一结果就是“100”、“1000”这样比原来多一位的。
则新建一个比原数组长度多一的数组,并且将数组第0个赋值为1返回即可。

实现:


class Solution {
    public int[] plusOne(int[] digits) {
        int n = digits.length;
        for(int i = n-1; i >= 0; i --) {
            if(digits[i] < 9) {
                digits[i]++;
                return digits;
            }
            digits[i] = 0;
        }
        int[] result = new int[n+1];
        result[0] = 1;
        return result;
    }
}