[LeetCode/Python/Java]Array and String - Plus One

문제

Given a non-empty array of digits representing a non-negative integer, increment one to the integer.

The digits are stored such that the most significant digit is at the head of the list, and each element in the array contains a single digit.

You may assume the integer does not contain any leading zero, except the number 0 itself.

 

이 문제는 입력된 배열을 한 개의 숫자로 인식하여 1로 더한 후의 값을 배열로 출력해야 합니다.

예를 들어서 입력 배열이 [1, 2, 3]일 경우 123에 1을 더한 124의 값을 [1, 2, 4]로 출력해야 합니다.

 

코드를 보시려면 더보기를 클릭해 주세요.

파이썬 코드

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        answer = []
        pre = 1
        for i in digits[::-1]:
            if i + pre == 10:
                answer.append(0)
            else:
                answer.append(i+pre)
                pre = 0
        if pre == 1:
            answer.append(1)
        return answer[::-1]

 

자바 코드

처음 푼 코드로 시간이 동작 시간이 오래 걸립니다.

class Solution {
    public int[] plusOne(int[] digits) {
        int pre = 1;
        for (int i=digits.length-1; i>-1; --i){
            digits[i] += pre;
            if (digits[i] == 10){
                digits[i] = 0;
            }else{
                pre = 0;
                break;
            }
        }
        if (pre == 1){
            int[] answer = new int[digits.length+1];
            answer[0] = pre;
            for (int i=1; i<digits.length+1; ++i){
                answer[i] = digits[i-1];
            }
            return answer;
        }else{
            return digits;
        }
    }
}

다른 사람의 코드로 동작 시간이 빠릅니다.

class Solution {
    public int[] plusOne(int[] digits) {
        List<Integer> list = new ArrayList<>();
        int ten = 1;
        for(int i=digits.length-1;i>=0;i--){
            int n = digits[i];
            int sum = n + ten;
            
            list.add(sum%10);
            ten = sum / 10;
            
        }
        if(ten == 1) list.add(ten);
        
        
        Collections.reverse(list);
        
       int[] ret = new int[list.size()];
        for (int i=0; i < ret.length; i++)
        {
            ret[i] = list.get(i).intValue();
        }
        return ret;
    }
}