给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3] 输出: [1,2,4] 解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1] 输出: [4,3,2,2] 解释: 输入数组表示数字 4321。
解析:
该题目主要考虑加一之后的进位问题,所以需要设置一个变量保存是否有进位信息。如果最后一位没有进位信息,则整个数组都不可能有进位,所以,此时直接返回原数组最后一位加一就可以了;如果最后一位出现了进位,则循环查看其他位是否有进位,如果有进位,则继续循环,否则将不再有进位,直接返回就可以了。如果到最高位有进位,则需要在数组最前面补一个1,可以使用vector的insert方法。
代码:
vector<int> plusOne(vector<int>& digits)
{
int size = digits.size();//数组大小
int sum = digits[size - 1] + 1;
if (sum < 10)//最后一位是否有进位
{
digits[size - 1] = sum;
return digits;
}
int carry = 1;//进位的标志
digits[size - 1] = 0;
for (int i = size - 2; i >= 0; i--)
{
if (digits[i] + carry >= 10)//如果有进位carry设为1,原值为0
{
digits[i] = 0;
carry = 1;
}
else//没有进位,则原值加carry,carry设为0,返回数组
{
digits[i] = digits[i] + carry;
carry = 0;
return digits;
}
}
if (carry == 1)//最高位还有进位,则补1
{
digits.insert(digits.begin(), 1);
}
return digits;
}