[Programmers 코딩 기초 트레이닝 : Java] 마지막 두 원소

🔗 문제 링크

프로그래머스 181927 - 마지막 두 원소

---

💡 문제 요약

정수 배열 num_list가 주어집니다.

• 마지막 원소(num_list[-1])가 그 이전 원소(num_list[-2])보다 크면
  → 마지막 원소에 두 원소의 차를 추가
• 그렇지 않으면
  → 마지막 원소에 2를 곱한 값을 추가
• 새로운 배열을 반환합니다.

예시
num_list = [2, 1, 6]
마지막 두 수: 1, 6 → 6 > 1 → 차이 5
결과 = [2, 1, 6, 5]

---

🧩 내가 작성한 코드

 

 

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

🧐 분석

• 배열을 새로 만들고, 직접 for문으로 기존 배열의 값을 복사
• 작동은 하지만 코드가 장황함
• 자바에서는 이런 단순 복사를 Arrays.copyOf()로 한 줄에 처리할 수 있음

---

✅ 개선한 코드

 

import java.util.Arrays;

class Solution {
    public int[] solution(int[] num_list) {
        
        int numListLen = num_list.length;
        
        int[] answer = Arrays.copyOf(num_list, numListLen+1);
        
        if(num_list[numListLen-1] > num_list[numListLen-2]) {
            answer[numListLen] = num_list[numListLen-1] - num_list[numListLen-2];
        } else {
            answer[numListLen] = num_list[numListLen-1] * 2;
        }
        
        return answer;
        
    }
}

---

⚡ 개선 포인트

항목기존 코드개선 코드

배열 복사	직접 for문 작성	Arrays.copyOf() 한 줄
가독성	낮음	매우 높음
성능	거의 동일 (O(n))	동일 (내부적으로 System.arraycopy 사용)
코드 길이	길고 반복적	간결하고 직관적

🧠 핵심 배움 포인트

• Arrays.copyOf(original, newLength)는
  → 기존 배열을 새 길이로 복사하면서 새 배열 반환
• 내부적으로 System.arraycopy()를 사용하므로 빠르고 안전
• 배열에 새 요소를 추가할 때 자주 쓰이는 표준적 방법

---

✍️ 한줄 요약

“직접 복사보다는 표준 라이브러리를 적극 활용하자.
Arrays.copyOf()는 코드 가독성과 유지보수를 크게 향상시킨다.”