题目描述:
Given a non-empty array of digits representing a non-negative integer, plus one to the integer.
The digits are stored such that the most significant digit is at the head of the list, and each element in the array contain a single digit.
You may assume the integer does not contain any leading zero, except the number 0 itself.
Example 1:
Input: [1,2,3]
Output: [1,2,4]
Explanation: The array represents the integer 123.
Example 2:
Input: [4,3,2,1]
Output: [4,3,2,2]
Explanation: The array represents the integer 4321.
要完成的函数:
vector<int> plusOne(vector<int>& digits)
说明:
1、这道题给定一个vector,里面装了几个个位数,这几个数形成了一个整数,比如[1,3,2,4],就是1324。要求把这个整数加一,然后把结果返回,还是返回这种格式的vector。
2、这道题很容易,我们最开始设定i=digits.size()-1,也就是最后一位,我们不断地判断digits[i]的数值是否为9,如果是,把这一位改为0,接着i--,继续判断前一位。
如果某一位不等于9了,那么我们结束循环,把这一位加一。如果一直跑完所有数值,都等于9,那么我们在最前面插入1。最后返回digits。
代码如下:
vector<int> plusOne(vector<int>& digits)
{
int i=digits.size()-1;
while(digits[i]==9&&i>=0)//如果当前位为9,并且i>=0
{
digits[i]=0;//把当前位修改为0
i--;//i=i-1
}
if(i==-1)//特殊状态,比如999这种
digits.insert(digits.begin(),1);
else//常见状态,比如299
digits[i]++;
return digits;
}
上述代码实测4ms,beats 99.68% of cpp submissions。