题目描述
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
假设除了整数 0 之外,这个整数不会以零开头。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/plus-one
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解答
class Solution(object):
def plusOne(self, digits):
"""
:type digits: List[int]
:rtype: List[int]
"""
for i in range(len(digits)-1, -1, -1):
if (digits[i]+1)%10 : # 末尾数加1再除以10有余数,即加1不进位
digits[i] = digits[i] + 1
return digits
else:
digits[i]=0
if digits[0]==0: # 最高位为0,即原数为9、99、999之类的
digits.insert(0,1)
return digits
倒序遍历列表,给末位加1。
如果末位加1不进位,则加1后返回;如果末位加1进位,则末位变为0,再向前一位加1;重复以上两步。
列表遍历完都没有返回,即每位加1都进位了(99,999,9999的情况),遍历完列表变为全0(00,000,0000),在最高位插入1。