Python으로 푸는 LeetCode 66. Plus One (Easy)

single digit(0~9)으로만 구성되어 있는 리스트를

마치 숫자처럼 생각하여 1을 증감시킬때,

리스트가 어떻게 변할지 프로그램을 짜시오.

LeetCode에서 푼 문제 리스트 보기

LeetCode에서 문제 보기

github에서 코드 보기

문제 풀이

single digit으로 구성된 리스트를 문자로 바꿔준 다음에 join() 메소드를 활용해서 문자열로 합쳐준다. 그리고 합쳐준 문자열을 숫자로 변환한 다음에 1을 더해준다. 더해준 숫자를 다시 문자열로 변환한 다음에 리스트로 만들어 반환해준다. 

파이썬 코드

class Solution:
    def plusOne(self, digits):
        """
        Runtime : faster than 61.90% of Python3
        Memory Usage : less than 5.29% of Python3
        """
        digits = list(map(str, digits))
        nums = int(''.join(digits))
        nums += 1
        nums = list(str(nums))
        return list(map(int, nums))
 
s = Solution()
print(s.plusOne([1,2,3]))
print(s.plusOne([4,3,2,1]))