[프로그래머스] 마지막 두 원소

문제

코드

Python

def solution(num_list):
    if(num_list[-1]>num_list[-2]):
        num_list.append(num_list[-1]-num_list[-2])
    else:
        num_list.append(num_list[-1]*2)
    return num_list

Java

class Solution {
    public int[] solution(int[] num_list) {
        int[] answer = new int[num_list.length + 1];
        int length = num_list.length;
        for(int i=0; i<num_list.length; i++){
            answer[i]=num_list[i];
        }
        if(num_list[length-1]>num_list[length-2]){
            answer[num_list.length]=num_list[length-1]-num_list[length-2];
        }
        else{
            answer[num_list.length]=num_list[length-1]*2;
        }
        return answer;
    }
}