LeetCode66.加一

原题链接：https://leetcode-cn.com/problems/plus-one/

66. 加一

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

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

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

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

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

思路

1.转换为数字，直接加一再转换为数组并输出（数组过大时会溢出）
2.直接在数组上操作，遇到进位则模拟进位过程；注意第一个数进位时要返回一个新数组（因为此时旧数组位数不够进位）

代码实现

    public static int[] plusOne(int[] digits) {
        if (digits[0] == 0) return new int[] {1};
        long val = 0;
        int length = digits.length;
        for (int digit : digits) {
            val = 10 * val + digit;
        }
        long newval = val+1;
        int count = 0;
        long temp = newval;
        while (temp!=0){
            temp/=10;
            count++;
        }
        int[] ans = new int[count];
        for (int i = ans.length-1; i >=0 ; i--) {
            ans[i] = (int) (newval%10);
            newval/=10;
        }


        return ans;
    }

public class Main {
    public static void main(String[] args) {
        int[] a = {1,2,3};
        int[] b = {9,9,9};
        int[] c = {0};
        int[] d = {9};
        int[] res = plusOne(d);
        System.out.println(Arrays.toString(res));
    }
    public static int[] plusOne(int[] digits) {
        if (digits[0] == 0) return new int[] {1};
        int length = digits.length;
        int lastnum = digits[length-1];
        if (lastnum+1 != 10){
            digits[length-1] = lastnum+1;
        }else {
            return carry(digits,length-1);
        }

        return digits;
    }

    private static int[] carry(int[] nums, int k){
        if (k<0) {
            int[] ans = new int[nums.length+1];
            ans[0] = 1;
            return ans;
        }
        nums[k]=0;
        if (k-1<0){
            return carry(nums,--k);
        }
        if (nums[k-1]+1 != 10){
            nums[k-1] = nums[k-1]+1;
            return nums;
        }else {
            return carry(nums,--k);
        }

    }

3.简洁版2

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

作者：guanpengchn
链接：https://leetcode-cn.com/problems/plus-one/solution/hua-jie-suan-fa-66-jia-yi-by-guanpengchn/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。