加一

题目

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

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

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

示例 1:

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

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

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/plus-one 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


解决方法

加一有两种情况:

  1. 数字9之外的数加一;
  2. 数字9加一;

计算也分为两种情况:

  1. 数字9之外加一,直接返回结果就行。
  2. 数字9加一,会产生进位,且进位为1。

所以计算加一可以从末尾遍历数组,没有进位就返回结果,有进位就循环计算直至遍历结束。

还有一种特殊情况,输入位全为9的时候,这时手动进位。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
public class PlusOne{
    public int[] plusOne(int[] digits) {
        for (int i = digits.length - 1; i >= 0; i--){
            digits[i]++;
            digits[i] = digits[i] % 10;
            if (digits[i] != 0){
                return digits;
            }
        }
        digits = new int[digits.length + 1];
        digits[0] = 1;
        return digits;
    }
}