leetcode 66 加一

题目描述

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

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

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

示例 1:

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

示例 2:

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

---

解决思路：

• 首先判断数组中有多少个9， 如果全是9，那么就需要创建一个 比原数组大 1 的新数组用于存放结果；
• 然后从下标最大处开始循环， 如果当前是下标最大处就进行 +1 运算，并判断是否需要进位，如需要 needAdd = 1， 否则needAdd = 0, 如果不是下标最大处，直接判断进位needAdd是否为1，如果为 1 就进行 和运算， 否则 直接保存当前下标处值；
• 最后，如果是全为9， 就会出现长度越位， 那么需要在程序最后判断进位是否为1，即 needAdd = 1？ 如果 为 1， 就将最高位，即下标最小处 置为 1。

执行用时 :1 ms, 在所有 Java 提交中击败了96.75%的用户
内存消耗 :35.5 MB, 在所有 Java 提交中击败了36.86%的用户

资源耗费

时间	内存
1ms	35.3MB

程序代码

class SolutionX{
 public int[] plusOne(int[] digits) {
     int needAdd = 0;
     int nineNums = 0;
     int needLength = digits.length;
     int []resList;
     int indexRight = 0;
     for(int i = 0; i < digits.length; i++){
         if (digits[i] == 9){
             nineNums++;
         }
     }
     if(nineNums == digits.length ){
         needLength = digits.length + 1;
         indexRight = 1;
     }
     resList = new int[needLength];
     
     for(int j = digits.length - 1; j >= 0; j--){
         int ele = 0;
         if(digits.length - 1 == j){
             if(digits[j] + 1 == 10){
                 resList[j+indexRight] = 0;
                 needAdd = 1;
             }else{
                 resList[j+indexRight] = digits[j] + 1 ;
                 needAdd = 0;
             }
         }else{
             if(needAdd == 1){
                 if(digits[j] + 1 == 10){
                     resList[j+indexRight] = 0;
                     needAdd = 1;
                 }else{
                     resList[j+indexRight] = digits[j] + 1 ;
                     needAdd = 0;
                 } 
             }else{
                 resList[j+indexRight] = digits[j];
             }
         }
     }
     if(needAdd == 1){
         resList[0] = 1;
     }
     
     return resList;

 }

 
}

public class Solution{
 public static void main(String[] args) {
     int []arr = new int[]{4,3,2,1};
     SolutionX sx = new SolutionX();
     int []res = sx.plusOne(arr);
     for(int i=0;i<res.length; i++){
         System.out.print(res[i]+" ");
     }
 }
}

上面代码是我一开始就进行全局考虑的后果，导致代码太长，没啥优点。下面是比较简单的解决方法。

思路： 基于数组本身进行加和操作， 最后在判断是否需要最高位需要置 1 否。

class Solution {
    public int[] plusOne(int[] digits) {
        for (int i = digits.length - 1; i >= 0; i--) {
            digits[i]++; // 此处的++ 既包含了 一开始的最低位 +1 还包括 每次的进位 +1
            digits[i] = digits[i] % 10; // 对当前位进行 取余数操作 保存个位数值 并以此判断是否进位
            if (digits[i] != 0) return digits; // 如果 digits[i] == 0 说明 之前的+1没有产生进位 循环结束 否则前一位继续 ++
        }
        digits = new int[digits.length + 1]; // 如果到这一步仍没有 返回 就说明 可能全是 9 那么就 说明 最终结果 长度比原来 多 1 也只有最高位是 1 其余全是 0
        digits[0] = 1;
        return digits;
    }
}

作者：yhhzw
链接：https://leetcode-cn.com/problems/plus-one/solution/java-shu-xue-jie-ti-by-yhhzw/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。