LeetCode:加一
给定一个非负整数组成的非空数组,在该数的基础上加一,返回一个新的数组。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
程序如下:
#include <iostream>
using namespace std;
#include <vector>
vector<int> plusOne(vector<int>& digits) {
int len = digits.size();
for (int i = len - 1; i >= 0; i--)
{
//如果当前元素不是9的情况
if (digits[i] < 9)
{
//当前元素加一:一种情况是刚开始的加一,另一种是来自于低位的进位
digits[i]++;
//由于当前元素加一之后,不会向高位进位,所以直接返回
return digits;
}
//如果当前元素时9,则置当前元素为0,向前进1
digits[i] = 0;
}
//当全部是9的时候,必然位数会增加一位,且首位是1
vector<int> answer(len + 1);//不赋值,所有默认为0
answer[0] = 1;
return answer;
}
int main()
{
vector < int > v1 = { 1,8 ,9,9 };
vector < int > v2 = { 9, 9 };
vector<int> nv1 = plusOne(v1);
vector<int> nv2 = plusOne(v2);
for (unsigned i = 0; i < nv1.size(); i++)
cout << nv1[i] << " ";
cout << "\n";
for (unsigned i = 0; i < nv2.size(); i++)
cout << nv2[i] << " ";
return 0;
}
结果如下:
1 9 0 0
1 0 0