加一问题
问题描述:将非负数组看数加一然后返回数组
比如{1,2,3}——{1,2,4}
{7,9,9}——{8,0,0}
那我们首先的思路是不是将数组变为一个数,加上1再返回数组里面。就像这样的程序:
class Solution
{
public:
vector<int> plusOne(vector<int>& digits)
{
vector <int> num;
int result = 0;
for(int num:digits)
{
result = result * 10 + num;
}
result += 1;
while (result > 0)
{
int kt = result % 10;
num.insert(num.begin(), kt);
result /= 10;
}
return num;
}
};
很可惜这个代码虽然思路粗暴但却有致命弱点——数大溢出,处理大数时可能会遇到溢出的问题。这就行不通了,我们改变下思路, 分析一下可能出现的情况:
1.例如{1,2,3}这种不用进位的情况,直接在末尾加一即可;
2.如{8,9,9}这种需要连续进位的情况,这种需要在进位左侧加一,变成{9,0,0},而且决定了我们只能从后向前遍历;
3.如{9,9}这种需要增容的情况变为{1,0,0}。
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
vector<int> plusOne(vector<int>& digits)
{
for (int i = digits.size() - 1; i >= 0; i--) {
if (digits[i] == 9) {
digits[i] = 0;
}
else {
digits[i] += 1;
return digits;
}
}
//如果所有位都是进位,则长度+1
digits.push_back(0);
digits[0] = 1;
return digits;
}
};
int main()
{
Solution st;
vector<int> a = { 9,9,9 };
vector<int> result = st.plusOne(a);
for (int num : result)
{
cout << num << " ";
}
return 0;
}