加一 LeetCode Golang/C++ 实现方法

GO:
func plusOne(digits []int) []int {
lens := len(digits)
for i:= lens-1;i>=0;i–{
if digits[i]==9{
digits[i]=0
}else{
digits[i]=digits[i]+1
return digits
}
}
if digits[0]==0{
digits[0]=1
digits = append(digits,0)
}
return digits
}
///////////////////////////////////////////////

C++：
vector plusOne(vector& digits) {
int n = digits.size();
for(int i=n-1;i>=0;i–)
{
if(digits[i]==9)
{
digits[i]=0;
}
else
{
digits[i]+=1;
return digits;
}
}
if(digits.front()==0)
{
digits.insert(digits.begin(),1);
}
return digits;

}