【每日刷题】加一

66. 加一 - 力扣（LeetCode）

方法一

这版是自己的AC代码，记录一下。

首先用flag表示此刻是否有进位（当前位是否需要+1），初始置true。先假设最后答案数组位数与原数组位数相同，如果flag为真，说明当前位需要+1，即 res[i] = digits[i] + 1 ，如果+1后 res[i] == 10 ，说明需要进位，把res[i]置为0，flag置为true。否则res[i]不需要有变化，flag置为false。

最后判断结束后flag。如果为false直接返回res就可以了。但如果依然为true就说明一直都有进位，就是9999这种数。最开始我在if(flag)里面添加了每一位赋值的操作，让 newRes[i+1] = res[i] 。后面看了别人的题解，反应过来如果flag为真，说明答案是100000这种，末尾肯定都是0。所以直接给newRes[0]赋值为1即可，其它位都是初始化的0，就可以直接返回了。

class Solution {public int[] plusOne(int[] digits) {int len = digits.length;boolean flag = true;int res[] = new int[len];for(int i = len - 1; i >= 0; i--) {if(flag) {res[i] = digits[i] + 1;if(res[i] == 10){res[i] = 0;flag = true;} else{flag = false;}} else {res[i] = digits[i];}}if(flag) {int newRes[] = new int[len + 1];newRes[0] = 1;return newRes;} else{return res;}}
}

方法二

评论区找到的最满意的一版题解。

直接判断加1前digits[i]是否为9，是的话就让它为0，继续循环。如果不为9，这里加上1后就可以结束了，直接返回digits数组。如果for循环结束还没有返回，说明给的数是9999这种一直走的if语句，那么就给数组扩容1个数并且重新初始化，把首位置为1，返回即可。

class Solution {public int[] plusOne(int[] digits) {int len = digits.length;for(int i = len - 1; i >= 0; i--) {if(digits[i] == 9) {digits[i] = 0;} else{digits[i]++;return digits;}}digits = new int[len + 1];digits[0] = 1;return digits;}
}