描述:
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;
}