[프로그래머스] JAVA 0단계 마지막 두 원소

마지막 두 원소

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

 

풀이

import java.util.*;

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

 

결과

테스트 1
입력값 〉	[2, 1, 6]
기댓값 〉	[2, 1, 6, 5]
실행 결과 〉	테스트를 통과하였습니다.

테스트 2
입력값 〉	[5, 2, 1, 7, 5]
기댓값 〉	[5, 2, 1, 7, 5, 10]
실행 결과 〉	테스트를 통과하였습니다.