LeetCode刷题之加一问题

108 阅读1分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。


题目描述

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

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

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

示例1 :

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

示例2:

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

思路:模拟+遍历

分析:因为数组的最后一位是个位,所以从个位+1判断是不是10的倍数,如果不是,则该位+1就是该位的结果,因为该位不能进位,所以之后都不用判断了,如果某一位加1是10的倍数,则该位变为0,计算下一位,如果最高位也变成0时,证明该数组之前每位都是9,则建立新的数组,并把首位赋值为1即是计算结果

class Solution {
    public int[] plusOne(int[] digits) {
       if(digits == null || digits.length <= 0){
            return digits;
        }
        int len = digits.length;
        for (int i = len-1; i >= 0; i--) {
            int num = digits[i]+1;
            if(num%10 > 0){
                digits[i] = num;
                break;
            }else{
                digits[i] = 0;
                if(i == 0){
                    digits = new int[len+1];
                    digits[i] = 1;

                }
            }

        }
        return digits; 
    }
}

思路2:递归

    public int[] plusOne(int[] digits) {
        return reversePlusOne(digits.length - 1, digits);
    }

    public int[] reversePlusOne(int index, int[] digits) {
        if (index < 0) {
            return digits;
        }
        digits[index]++;
        digits[index] %= 10;
        if (digits[index] != 0) {
            return digits;
        } else if (index == 0) {
            digits = new int[digits.length + 1];
            digits[0] = 1;
            return digits;
        }
        return reversePlusOne(index - 1, digits);
    }