【LeetCode 66】加一

版权声明：本文为博主原创文章，转载请注明出处，谢谢大家O(∩_∩)O https://blog.csdn.net/Stripeybaby/article/details/84172721

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

代码实现

package com.bupt.wwy;

public class Main {

    public static void main(String[] args) {
	// write your code here
        int[] arr = {1, 2, 3};
        Solution s = new Solution();
        int[] res = s.plusOne(arr);
        for (int i=0;i<res.length;i++){
            System.out.print(res[i]);
        }
    }
}

class Solution{
    public int[] plusOne(int[] digits){
        boolean flag = true;
        for (int i=0;i<digits.length;i++){
            if (digits[i] != 9){
                flag = false;
                break;
            }// 只要有一位不为9，那么数组的长度就不会改变
        }
        if (flag){ // 数组元素都为9，数组长度会增加一位
            int[] result = new int[digits.length+1];
            result[0] = 1;// 将新数组的首元素设置为1，其余位置默认为0
            return result;
        }else {
            for (int i=digits.length-1;i>=0;i--){
                if (digits[i]!=9){
                    digits[i]+=1;
                    break;
                }else {
                    digits[i]=0;
                }
            }
        }
        return digits;
    }
}