Leetcode: Plus One

全栈程序员站长

发布于 2022-07-11 09:29:02

4190

发布于 2022-07-11 09:29:02

文章被收录于专栏：全栈程序员必看全栈程序员必看

大家好，又见面了，我是全栈君。

题目： Given a non-negative number represented as an array of digits, plus one to the number.

The digits are stored such that the most significant digit is at the head of the list.

题目意思是说给定一个数这个数以数组的形式表示，就是说这个数各个位上的数保存在一个数组里面，然后给这个数加一。求返回的结果。当然仍旧以数组的形式返回。

C++演示样例代码：

class Solution {
public:
    vector<int> plusOne(vector<int> &digits)
    {
        vector<int>::size_type size = digits.size();
        //当digits的每一位都是9的时候，由于返回的结果位数要添加1。所以做一个特殊处理。

我写的代码看起来比較繁琐，看到网上有高手这样做的：

class Solution 
{
public:
    vector<int> plusOne(vector<int> &digits) 
    {
        vector<int> ret(digits);
        reverse(ret.begin(), ret.end());

        int flag = 1;
        for(int i = 0; i < ret.size(); i++)
        {
            ret[i] += flag;
            //这里的flag的结果要么是0。要么是1
            flag = ret[i] / 10;
            ret[i] %= 10; 
        }

        if (flag == 1)
            ret.push_back(1);

        reverse(ret.begin(), ret.end());

        return ret;
    }
};

只是，最后这两段代码咱Leetcode上提交以后都用时是4ms！肯能是后者的代码中用了两次vector的reverse略微占了一些时间吧。

发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/115231.html原文链接：https://javaforall.cn

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2022年2月7，如有侵权请联系 cloudcommunity@tencent.com 删除

编程算法