leetcode066-python 加一

最新推荐文章于 2022-01-21 20:38:39 发布

原创最新推荐文章于 2022-01-21 20:38:39 发布 · 215 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode-简单

leetcode 专栏收录该内容

25 篇文章

订阅专栏

本文介绍了一种针对非空整数数组的加一操作算法，该数组表示一个非负整数，通过对原始复杂实现的逐步简化，最终形成简洁高效的解决方案。文章通过两个示例展示了如何将数组中的每个数字加一，并处理进位情况。

Given a non-empty array of digits representing a non-negative integer, plus one to the integer.

The digits are stored such that the most significant digit is at the head of the list, and each element in the array contain a single digit.

You may assume the integer does not contain any leading zero, except the number 0 itself.

Example 1:

Input: [1,2,3]
Output: [1,2,4]
Explanation: The array represents the integer 123.

Example 2:

Input: [4,3,2,1]
Output: [4,3,2,2]
Explanation: The array represents the integer 4321.

解题要点：

1.如果遇到9就进一位，但是写的太复杂了

class Solution:
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """
        if digits == [9]:
            return [1,0]
        n = len(digits)-1
        if digits[n]!=9:
            digits[n] += 1
            return digits
        else:
            while n>0 :
                digits[n] = 0
                digits[n-1] += 1
                if digits[n-1]!=10:
                    break
                n = n - 1
            if digits[0] ==10:
                digits[0] = 0
                digits.insert(0, 1)
            return digits

后来借鉴改进后，取用了一个进位量ca：

class Solution:
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """
        n = len(digits)-1
        digits[n] += 1
        ca = 0
        while n >= 0:
            digits[n] += ca
            if digits[n] > 9:
                ca = 1
                digits[n] = 0
            else:
                ca = 0
                break
            n = n - 1
        if ca != 0:
            digits.insert(0, 1)
        return digits