Plus One

Key Idea

Solution

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        # Inputs: List[int]
        # Outputs: List[int]
        # Description:
            # Increment "digits" by one and return the list
        # E.g. [1,2,3] reutrns [1,2,4]
        
        # Increment last by 1
        for i in range(len(digits) - 1, -1, -1):
            if digits[i] < 9:
                digits[i] += 1
                return digits
            digits[i] = 0
        return [1] + digits

Complexity