Leetcode: Plus One

来源:互联网 发布:aes算法的原理和算法 编辑:程序博客网 时间:2024/05/16 11:03

题目:
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,所以做一个特殊处理。        bool flag = true;        for (size_t i = 0; i < size; i++)        {            if (9 != digits[i])            {                flag = false;            }        }        if (flag)        {            digits.resize(size + 1);            digits[0] = 1;            for (size_t i = 1; i <= size; i++)            {                digits[i] = 0;            }            return digits;        }        //当digits的每一位并不都是9的时候        //如果最后一位不是9,直接加一返回        if (digits[size - 1] != 9)        {            digits[size - 1]++;            return digits;        }        //如果最后一位是9,则变成0,前面一位加一,如果加一以后变成10,则循环进行这样的处理        else        {            digits[size - 1] = 0;            digits[size - 2]++;            for (int i = size - 2; i >= 0; i--)            {                if (10 == digits[i])                {                    digits[i] = 0;                    digits[i - 1]++;                }            }        }        return digits;    }};

我写的代码看起来比较繁琐,看到网上有高手这样做的:

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稍微占了一些时间吧!

0 0
原创粉丝点击