문제/릿코드

[릿코드/LeetCode] 66. Plus One (파이썬3/Python3)

개 살구 2021. 7. 8. 00:23

문제

https://leetcode.com/problems/plus-one/ 

 

Plus One - LeetCode

Level up your coding skills and quickly land a job. This is the best place to expand your knowledge and get prepared for your next interview.

leetcode.com


생각

숫자(리스트)를 count up을 한 번 하는 것이 목표

마지막 자리를 +1하면 되지만 carry가 발생하는 경우를 생각해줘야함.

가장 마지막 원소를 +1하고 carry가 발생하면 앞쪽으로 넘겨주기?


코드

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        digits[-1] += 1
        
        for i in range(len(digits)-1, 0, -1):
            if digits[i] >= 10:
                digits[i] %= 10
                digits[i-1] += 1
            
        if digits[0] >= 10:
            digits[0] %= 10
            digits.insert(0, 1)
            
        return digits

for문 쓰면서 괄호 쓴 거... 장난 아니네...

헤헤