leetcode_66_ 加一

题目大意

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

思路分析

整个的难点就是进位的问题，也就是讨论9出现的位置

1. 当最后一位不是9时，加一返回
2. 当后几位是9，寻找中间不是9的数加一，后几位变0返回
3. 当所有数都是9时，我们只需要构造一个长度比原数组长度+1的新数组，将首元素置为 1，其余元素置为 0 即可

当然也可以直接关注在加一后是否为0，代码如下

代码

class Solution {
    public int[] plusOne(int[] digits) {
        int n = digits.length;
        // 考虑 9 出现的位置
        for (int i = n - 1; i >= 0; --i) {
            if (digits[i] != 9) {
                ++digits[i];
                for (int j = i + 1; j < n; ++j) {
                    digits[j] = 0;
                }
                return digits;
            }
        }
        // 考虑 0 的情况
        for (int i = n - 1; i >= 0; i--) {
           digits[i] = (digits[i] + 1) % 10;
           
           if (digits[i] != 0) 
               return digits;
       	}


        // digits 中所有的元素均为 9
        int[] ans = new int[n + 1];
        ans[0] = 1;
        return ans;
    }
}