[leetcode.com]算法题目 - Plus One

Given a number represented as an array of digits, plus one to the number.

class Solution {
public:
vector<int> plusOne(vector<int> &digits) {
// Start typing your C/C++ solution below
// DO NOT write int main() function
bool allNine = true;
int size = digits.size();

for(int i = 0; i< size; i++){
if(digits[i] != 9){
allNine = false;
break;
}
}

if(allNine){
vector<int> result(1+size, 0);
result[0] = 1;
return result;
}

vector<int> result(size);

for(int i=0;i<size;i++){
result[i]=digits[i];
}

result[size-1] += 1;
int k = size-1;

while(10==result[k]){
result[k] = 0;
k--;
result[k]++;
}

return result;
}
};

我的答案
思路：可能出现的意外情况只有数字全是9的时候，这种情况单独拿出来讨论一下，剩下的情况都不可能全为9了。