简单题66-加一（Python）20240918

问题描述：

python

class Solution(object):
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """
        n = len(digits)
        
        # 从最后一位开始处理进位
        for i in range(n - 1, -1, -1):
            if digits[i] < 9:
                digits[i] += 1
                return digits
            digits[i] = 0
        
        # 如果所有的数字都是 9，处理进位情况
        return [1] + digits

java

class Solution {
    public int[] plusOne(int[] digits) {
        int n = digits.length;  // 获取数组长度
        
        // 从最后一位开始向前遍历
        for (int i = n - 1; i >= 0; i--) {
            if (digits[i] < 9) {
                digits[i] += 1;  // 如果当前位小于 9，直接加 1
                return digits;   // 直接返回结果
            }
            digits[i] = 0;  // 如果当前位是 9，则将其设置为 0
        }
        
        // 如果所有位都是 9，则需要在数组前面添加 1
        int[] newDigits = new int[n + 1];  // 创建新数组
        newDigits[0] = 1;  // 第一位设置为 1，其余默认为 0
        return newDigits;
    }
}