给出一个数组代表一个数字,返回+1之后的数,用数组表示
public class Solution {
public int[] plusOne(int[] digits) {
boolean flag = true;
for(int i = digits.length - 1; i >= 0; --i){
digits[i] += (flag ? 1 : 0);
flag = digits[i] / 10 == 1;
digits[i] %= 10;
}
if(flag){
int[] ans = new int[digits.length + 1];
for(int i = 0; i < digits.length; ++i){
ans[i + 1] = digits[i];
}
ans[0] = 1;
return ans;
}
else return digits;
}
}