Leetcode: Plus One

全栈程序员-用户IM • 2022年2月7日上午11:00 • 未分类

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

题目：
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;
    }
};