題目描述:
給定一個由整數組成的非空數組所表示的非負整數,在該數的基礎上加一。
最高位數字存放在數組的首位, 數組中每個元素只存儲單個數字。
你可以假設除了整數 0 之外,這個整數不會以零開頭。
示例 1:
輸入: [1,2,3] 輸出: [1,2,4] 解釋: 輸入數組表示數字 123。
示例 2:
輸入: [4,3,2,1] 輸出: [4,3,2,2] 解釋: 輸入數組表示數字 4321。
思路:
考慮以下幾種數字,
9,99,999,99... 該類數字 + 1 後 數組會溢出,涉及數組擴容,擴容後新數組長度比原數組長1 ,數組爲對象,int類型初始化爲0,故此時只要將res[0] = 1,即可輸出。
19,139,899,該類數字... 將9變爲0,前一位數字+1即可。
112,其他數字...末位位+1即可。
代碼展示(Java)
class Solution {
public int[] plusOne(int[] digits) {
boolean flag = false;
for(int i : digits){
if(i!=9) {
flag = true;
break;
}
}
if(flag == false){
int[] res = new int[digits.length+1];
res[0]=1;
return res;
}
else{
for(int i = digits.length - 1;i>=0;i--){
if(digits[i]!=9){
digits[i]+=1;
break;
}
else{
digits[i]=0;
}
}
return digits;
}
}
}