https://leetcode.cn/problems/plus-one/description/
給定一個由 整數 組成的 非空 數組所表示的非負整數,在該數的基礎上加一。
最高位數字存放在數組的首位, 數組中每個元素只存儲單個數字。
你可以假設除了整數 0 之外,這個整數不會以零開頭。
示例 1:
輸入:digits = [1,2,3]
輸出:[1,2,4]
解釋:輸入數組表示數字 123。
示例 2:
輸入:digits = [4,3,2,1]
輸出:[4,3,2,2]
解釋:輸入數組表示數字 4321。
示例 3:
輸入:digits = [0]
輸出:[1]
提示:
1 <= digits.length <= 100
0 <= digits[i] <= 9
解答
這個算是高精度加法的簡化版本
我們在最末尾加上1 若正處理的該位超過10,該位編程個位並進位。 若該位小於10,則該位直接該位計算結果。
最後注意邊界, 若所有字符串計算完 還有進位則最高位還要再額外添加一位(999+1 = 1000 輸入的是999三位數字,結果爲1000四位數字)。
代碼如下
class Solution {
public:
vector<int> plusOne(vector<int>& d) {
int prev=1;
for(int i = d.size()-1;i>=0;i--){
int val = d[i];
val = val+prev;
if(val>=10){
d[i]=val%10;
prev = val/10;
}else{
d[i]= val;
prev=0;
}
}
if(prev!=0){
d.insert(d.begin(),1);
}
return d;
}
};