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

문제 설명

정수 리스트 num_list가 주어질 때, 마지막 원소가 그전 원소보다 크면 마지막 원소에서 그전 원소를 뺀 값을 마지막 원소가 그전 원소보다 크지 않다면 마지막 원소를 두 배한 값을 추가하여 return하도록 solution 함수를 완성해주세요.

 

제한사항

• 2 ≤ num_list의 길이 ≤ 10
• 1 ≤ num_list의 원소 ≤ 9

 

입출력 예

 

코드

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

 

문제 출처

https://school.programmers.co.kr/learn/courses/30/lessons/181927