題目描述
鏈接:https://leetcode-cn.com/problems/plus-one/
給定一個(gè)由 整數(shù) 組成的 非空 數(shù)組所表示的非負(fù)整數(shù)掂恕,在該數(shù)的基礎(chǔ)上加一渐溶。
最高位數(shù)字存放在數(shù)組的首位套利, 數(shù)組中每個(gè)元素只存儲(chǔ)單個(gè)數(shù)字及汉。
你可以假設(shè)除了整數(shù) 0 之外提佣,這個(gè)整數(shù)不會(huì)以零開頭
示例
輸入:digits = [1,2,3]
輸出:[1,2,4]
解釋:輸入數(shù)組表示數(shù)字 123。
輸入:digits = [1,2,9]
輸出:[1,3,0]
解釋:輸入數(shù)組表示數(shù)字 129梧宫。
代碼
public int[] plusOne(int[] digits) {
int flag = 0;
for(int i = digits.length - 1; i >= 0; i--) {
int sum = digits[i] + flag;
if (i == digits.length - 1) {
sum = sum + 1;
}
digits[i] = sum % 10;
flag = sum / 10;
}
if (flag <= 0) {
return digits;
}
// digits 中所有的元素均為 9
int[] ans = new int[digits.length + 1];
ans[0] = 1;
return ans;
}