💻 자바 알고리즘 - 마지막 두 원소

📕 문제 설명

정수 리스트 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];
        }
        answer[num_list.length] = (num_list[num_list.length-1]>num_list[num_list.length-2]) ? num_list[num_list.length-1]-num_list[num_list.length-2] : num_list[num_list.length-1]*2;
        return answer;
    }
}

이 문제의 핵심은 num_list의 마지막 원소다. 마지막 원소의 인덱스는 그 배열의 길이에서 -1을 한 값이므로 num_list[num_list.length-1]을 통해 마지막 원소를 가져올 수 있다.

 

먼저 for문을 돌면서 answer 배열에 num_list 원소를 복사한다.

그리고 삼항 연산자를 통해 마지막 원소가 그 이전 원소보다 크다면 마지막 원소 - 그 이전 원소 값을 answer 배열의 마지막 원소로 넣고 반대로 마지막 원소가 작다면 마지막 원소의 두배 값을 answer 배열에 넣는다.

 

💡 참고

• 배열.length : 배열 길이 리턴