题目描述:
66. 加一
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入:digits = [1,2,3]
输出:[1,2,4]
解释:输入数组表示数字 123。
示例 2:
输入:digits = [4,3,2,1]
输出:[4,3,2,2]
解释:输入数组表示数字 4321。
示例 3:
输入:digits = [0]
输出:[1]
开始没理解题目的意思,看实例要求还以为是直接末尾数字加一就好了。然后测试的时候出现了[9],需要变成[1,0]。
原始的错误代码,只能通过测试样例:
class Solution(object):
def plusOne(self, digits):
size = len(digits)
i = size - 1
digits[i] = digits[i] + 1
if digits[i] == 10 :
digits[i] = 1
digits = digits + [0]
return digits
return digits
看评论区,真正理解题目的意思,类似于一个数字的加法,需要进位这样的操作。
正确代码:
class Solution(object):
def plusOne(self, digits):
size = len(digits)
num = 0
for i in range(0,size):
num += digits[i] * pow(10, size - 1 - i)
num = num + 1
list = []
while num > 0:
list.append(num % 10)
num //= 10
list.reverse()
return list
#使用转换为列表的内置函数
#list = [int(x) for x in str(num)]