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

📎 문제 출처

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

📌 문제 설명

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

❓ 풀이 방법

num_list 보다 크기가 한 개 큰 배열을 생성한 후 num_list의 원소들을 복사합니다.
그 후 num_list의 마지막 두 개의 원소를 비교하여 if ~ else문으로 마지막 원소를 추가합니다.

💻 Code

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