LeetCode算法题66:加一解析

给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例1:

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。

示例2:

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

这个题分析起来一个比较直观的思路就是进位直接加和,但是要考虑两种情况,一种是位数不会发生变化,另一种就是类似999这类数字,一旦加一位数就会加一,所以首先判断是否所有数字都为9,如果是,则直接用一个长度比原来数组多一位的数组输出首位为1,其余位为0的数组代替原数组输出。否则就是位数不变,只需要考虑进位就可以,那么只要先在最后一位加1,然后反向遍历整个数组,如果当前位加上进位大于10产生了进位则继续向前遍历,并把当前位置为其对10的余数。如果没有产生进位直接退出遍历返回数组。
C++源代码:

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int num = 0, i = 0;
        int len = digits.size();
        for (i=0;i<len;i++)
        {
            if (digits[i]!=9)
                break;
        }
        if (i==len)
        {
            vector<int> result(len+1);
            result[0] = 1;
            for (int j=1;j<len+1;j++)
                result[j] = 0;
            return result;
        }
        int carry = 0;
        digits[len-1] += 1;
        for (i=len-1;i>=0;i--)
        {
            digits[i] += carry;
            carry = digits[i] / 10;
            digits[i] = digits[i] % 10;
            if(carry == 0) break;
        }
        return digits;
    }
};

python3源代码:

class Solution:
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """
        if len(set(digits))==1 and (9 in set(digits)):
            result = [1]
            for i in range(1,len(digits)+1):
                result.append(0);
            return result
        carry = 0
        digits[len(digits)-1] += 1
        for i in range(len(digits)-1, -1, -1):
            digits[i] += carry
            carry = digits[i] // 10;
            digits[i] = digits[i] % 10;
            if carry==0:
                break
        return digits

猜你喜欢

转载自blog.csdn.net/x603560617/article/details/83549799