Leetcode 66: Plus One

Question:

Given a non-negative number represented as an array of digits, plus one to the number.

The digits are stored such that the most significant digit is at the head of the list.

Subscribe to see which companies asked this question

 

博主工作了5个月，最近才又跑回来继续刷题，希望接下来能找到一份心仪的工作~

 

题目解析：

简单的一道对Array进行操作的题。重点是如何维护最后一位的进位问题。例如，给出的array是[9, 9, 9]的话，返回的结果是[1, 0, 0, 0]。

利用了一个flag来维护进位，等for循环结束的时候如果flag为1，那么最后一位需要被维护，也就新建一个数组，长度为原数组+1，把所有的值都存进去就行了。

 

public int[] plusOne(int[] digits) {
        int flag = 1;
        for(int i = digits.length - 1; i >= 0; i--){
            digits[i] += flag;
            if(digits[i] == 10){
                digits[i] = 0;
                flag = 1;
            }
            else if(digits[i] < 10){
                flag = 0;
            }
        }
        
        if(flag == 1){
            int[] newdigits = new int[digits.length + 1];
            newdigits[0] = 1;
            for(int i = 0; i < digits.length; i++){
                newdigits[i + 1] = digits[i];
            }
            return newdigits;
            
        }else{
            return digits;
        }
        
    }