給定一個由整數(shù)組成的非空數(shù)組所表示的非負整數(shù),在該數(shù)的基礎(chǔ)上加一。
最高位數(shù)字存放在數(shù)組的首位, 數(shù)組中每個元素只存儲一個數(shù)字。
你可以假設(shè)除了整數(shù) 0 之外,這個整數(shù)不會以零開頭。
示例 1:
輸入: [1,2,3]
輸出: [1,2,4]
解釋: 輸入數(shù)組表示數(shù)字 123。
示例 2:
輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入數(shù)組表示數(shù)字 4321。
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
vector<int> res=digits;
reverse(res.begin(),res.end());
vector<int> result;
int len=digits.size();
int carry,i=1;
int sum;
sum=res[0]+1;
carry=(res[0]+1)/10;
result.push_back(sum%10);
while(i<len)
{
int sum=res[i]+carry;
carry=sum/10;
result.push_back(sum%10);
i++;
}
if(carry)
result.push_back(1);
reverse(result.begin(),result.end());
return result;
}
};