프로그래머스 | 마지막 두 원소 JAVA (자바)

🧐 문제

: 마지막 두 원소

 

 

🧾 문제 설명

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

 

 

🤙🏻 제한 사항

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

 

 

🚩 입출력 예

1. 입력 : [2, 1, 6], 출력 : [2, 1, 6, 5]

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

 

2. 입력 : [5, 2, 1, 7, 5], 출력 : [5, 2, 1, 7, 5, 10]

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

 

 

---

✓ 문제 풀이

 

1. 크기가 1 더 큰 정수 리스트 list를 만든다.

2. 기존 배열의 크기를 n으로 받고 마지막 두 원소의 크기를 비교하여 새로운 원소 lastNum의 값을 정한다.

3. 새로운 배열에 값을 넣은 뒤 return한다.

 

import java.util.ArrayList;

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

 

✓ 다른 사람의 풀이

 

나와 전체적인 문제 흐름은 같고, 3항 연산자를 통해 if문을 사용하지 않고 마지막 값을 구했다는 점 정도가 다른 듯 했다.

 

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;
    }
}