描述:
Given a number represented as an array of digits, plus one to the number
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
思路:此题分两种情况讨论:
1,当数组最后一个数小于9时,我们直接将它加1就好了。
2,当最后一个数等于9时,这时候我们需要将它置为0,然后再循环遍历在它之前的数,看是否还存在9这个数,如果有就将它们置为0,没有的话就跳出循环,再判断此时的nIndex是否小于0,如果小于0,那说明所有的数字都是9,然后加1的结果就是增加一位数字,需要在数组头部插入1,反之,就将当前nIndex所在位置的那个数加1就好了。
#include <iostream>
#include <vector>
#include <iterator>
#include <algorithm>
using namespace std;
class solution{
public:
vector<int> plusOne(vector<int>& digits){
if(digits.empty())
{
return digits;
}
int nindex=digits.size()-1; //注意数组的下标是从0开始的
if(digits[nindex]<9)
{
digits[nindex]++;
}else //最后一位是9的情况
{
digits[nindex--]=0;
while(nindex>0 && digits[nindex]==9)
digits[nindex--]=0;
if(nindex<0)
{
digits.insert(digits.begin(),1);
}else{
digits[nindex]++;
}
}
return digits;
}
};
void display(const int& data){
cout<<data<<' ';
}
int main(int argc, char *argv[])
{
int a[]={9,9,9,9};
vector<int> vec(a,a+sizeof(a)/sizeof(int));
solution sl;
vector<int> myarr=sl.plusOne(vec);
for_each(myarr.begin(),myarr.end(),display);
return 0;
}