Lv0 - 마지막 두 원소

 

class Solution {
    public int[] solution(int[] num_list) {
        int count = num_list.length;
        int[] answer = new int[count + 1]; // 기존 배열 크기 + 1

        // 기존 배열 복사
        for (int i = 0; i < count; i++) {
            answer[i] = num_list[i];
        }

        // 새로운 숫자 추가
        if (num_list[count - 1] > num_list[count - 2]) {
            answer[count] = num_list[count - 1] - num_list[count - 2]; // 마지막 값 - 마지막 직전 값
        } else {
            answer[count] = num_list[count - 1] * 2; // 마지막 값 * 2
        }

        return answer;
    }
}

 

어차피 경우의 수는 마지막 원소가 전 원소보다 작거나 크거나 둘 중 하나이다.

그렇기때문에 이를 이용해 for, if문을 사용하여 해결하면 된다.