leetcode | 66.加一

66.加一

题目描述

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

示例 1：
输入：digits = [1,2,3]
输出：[1,2,4]
解释：输入数组表示数字 123。

题目链接

思路

1. 从后往前找第一个不为9的元素，并将该位置后的9置零
2. 将数组转化为数字加一再将结果转化为整数数组

   代码

   class Solution(object):
   	def plusOne(self, digits):
   		"""
   		从后往前找第一个不为9的元素，并将该位置后的9置零
           :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
   				for j in range(i + 1, n):
   					digits[j] = 0
   				return digits
   		return [1] + [0] * n
   	
   	def plusOne1(self, digits):
   		"""将数组转化为数字加一再将结果转化为整数数组"""
   		num = ""
   		for digit in digits:
   			num += str(digit)
   		return list([int(ch) for ch in str(int(num) + 1)])
   
   
   if __name__ == "__main__":
   	slt = Solution()
   	# result = slt.plusOne([1, 2, 3])
   	result = slt.plusOne1([1, 2, 3])
   	print(result)