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

sungeun·2023년 5월 26일
0

코딩 기초 트레이닝

목록 보기
26/124
post-thumbnail

📔 문제 설명

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

📌 제한사항

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

🔍 입출력 예

num_listresult
[2, 1, 6][2, 1, 6, 5]
[5, 2, 1, 7, 5][5, 2, 1, 7, 5, 10]

🔍 입출력 예 설명

입출력 예 #1

  • 마지막 원소인 6이 그전 원소인 1보다 크기 때문에 6 - 1인 5를 추가해 return합니다.

입출력 예 #2

  • 마지막 원소인 5가 그전 원소인 7보다 크지 않기 때문에 5의 두 배인 10을 추가해 return합니다.

🐹 내가 작성한 코드

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];
        }
        
        int last = num_list[num_list.length - 1];
        int before = num_list[num_list.length - 2];
        
        if (last > before) {
            answer[num_list.length] = last - before;
        } else {
            answer[num_list.length] = last * 2;
        }
        
        return answer;
    }
}
profile
Back-end Developer

0개의 댓글