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

描述:

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


 

发布了102 篇原创文章 · 获赞 26 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/weixin_42145502/article/details/104008204