leetcode第66題 加一(運行速度有點慢)

給定一個由整數組成的非空數組所表示的非負整數，在該數的基礎上加一。

最高位數字存放在數組的首位， 數組中每個元素只存儲單個數字。

你可以假設除了整數 0 之外，這個整數不會以零開頭。

示例 1:

輸入: [1,2,3]
輸出: [1,2,4]
解釋: 輸入數組表示數字 123。

示例 2:

輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入數組表示數字 4321。

// class Solution {
//     public int[] plusOne(int[] digits) {
//         LinkedList<Integer> res=new LinkedList<Integer>();
//         int add=1;
//         for(int i=digits.length-1;i>=0;i--){
//             if(add+digits[i]>=10&&add!=0){
//                 res.addFirst(add+digits[i]-10);
//                 add=1;
//             }else{
//                 res.addFirst(add+digits[i]);
//                 add=0;
//             }
//         }
//         if(add==1) res.addFirst(1);
//         int[] ret=new int[res.size()];
//         for(int i=0;i<ret.length;i++){
//             ret[i]=res.get(i);
//         }
//         return ret;
//     }
// }
class Solution {
    public int[] plusOne(int[] digits) {
        for (int i = digits.length - 1; i >= 0; i--) {
          if(digits[i]==9) digits[i]=0;
          else{
              digits[i]++;
              break;
          } 
        }
       if(digits[0]==0){
           int[] ans=new int[digits.length+1];
           ans[0]=1;
           return ans;
       }else{
           return digits;
       }
    }
}