【20190709】【每天一道算法题】加一（数组）

问题

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一，最高位数字存放在数组的首位， 数组中每个元素只存储一个数字。

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

输入: [1,2,3]，输出: [1,2,4]

解释: 输入数组表示数字 123。

示例 2:

输入: [4,3,2,1]，输出: [4,3,2,2]

解释: 输入数组表示数字 4321。

---

思路及解答

1. Python实现

# 方法一：类型转换（List -> int -> List）
# 将列表转换为数值，从最低位开始到最高位，分别乘以 10 的(-i-1)次幂，再相加。数值转换为列表时，用到的是整数除和取模。
# 时间复杂度：2*O(n)
class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        l = len(digits)
        num = 0
        resultList = []
        for i in range(-1, -l-1, -1):   
            num = num + digits[i] * pow(10, (-i - 1))
        result = num + 1
        while(result):
            resultList.append(result%10)
            result = result // 10
        return resultList[ : : -1]   # 列表倒序输出
		
		
# 方法二：逐位计算。
# 将resul初始化为列表最低位加一再模十（列表类型），若无进位carry初始化为0；若有进位carry初始化为1。在之后的循环中，位之和是当前位加上进位标志位再模十（carry要么为0，要么为1）；carry为当前位加上进位标志位再整除十。
# 时间复杂度：O(n)
class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        result = [(digits[-1] + 1) % 10]
		carry = (digits[-1] + 1) // 10
        for index in range(len(digits)-2, -1, -1):
            result.append((digits[index] + carry) % 10)
            carry = (digits[index] + carry) // 10
        if carry:        # 对于溢出位的处理，因为 carry 只能是 0 或 1，所以最后只需要在最高位补一即可。
            result.append(1)
            return result[: : -1]
        return result[: : -1]

---

知识点

1. 字符串 / 列表切片时间复杂度为：O(k)（也就是说不会超过O(n)）

参考：Python内置方法的时间复杂度

2. Python的内置函数

参考：Python内置函数详解——总结篇