Plus One

最新推荐文章于 2021-08-21 12:48:52 发布

phlovexz

最新推荐文章于 2021-08-21 12:48:52 发布

阅读量207

点赞数

分类专栏： LeetCode 文章标签： c语言

本文链接：https://blog.csdn.net/u013609078/article/details/51637079

版权

LeetCode 专栏收录该内容

109 篇文章 0 订阅

订阅专栏

c++

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        if (digits.empty())
            return digits;
        int update = 0;
        int n = digits.size() - 1;
        while (n>=0){
            if (n == digits.size() - 1)
                update += digits[n] + 1;
            else
                update += digits[n];
            digits[n] = update % 10;
            update /= 10;
            n--;
        }
        if (update > 0){
            digits.insert(digits.begin(), update);
        }
        return digits;
    }
};

python

class Solution(object):
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """
        if not digits: return digits
        update = 0
        n = len(digits)-1
        while n>=0:
            if n== len(digits)-1:
                update += digits[n]+1
            else:
                update += digits[n]

            digits[n] = update%10
            update /= 10
            n -= 1
        if update >0:
            return [1] + digits
        return digits