当前位置: 代码迷 >> 综合 >> C++ 笔试每日一题(leetcode Plus one)
  详细解决方案

C++ 笔试每日一题(leetcode Plus one)

热度:51   发布时间:2024-01-25 04:50:49.0

描述:

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;
}