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

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

처음에는 배열이기 때문에 습관적으로 for문에 넣었는데, 그냥 마지막 원소와 그 앞의 원소만 비교해 주면 되기 때문에 index로 바로 접근해 주면 된다. 대신 index는 0부터 시작이라는 점을 잘 활용해 주면 된다.

import java.util.*;
class Solution {
    public ArrayList solution(int[] num_list) {
        //마지막 원소 > 그전 원소 : 마지막 원소 - 그전 원소
        //마지막 원소 < 그전 원소 : 마지막 원소 * 2
        ArrayList<Integer> arr = new ArrayList<>();
        int length = num_list.length;
        for(int i=0; i<length; i++){
            arr.add(num_list[i]);
        }
        
        if(num_list[length-1] > num_list[length-2]){
            arr.add(num_list[length-1] - num_list[length-2]);
        }else{
            arr.add(num_list[length-1] * 2);
        }

        return arr;
    }
}