[Java] 코딩 39 - 프로그래머스 기출 문제(JAVA) / 두 개 뽑아서 더하기 / Day 34

📘문제 설명

정수 배열 numbers가 주어집니다. numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return 하도록 solution 함수를 완성해주세요.

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

프로그래머스

 

---

📌 제한조건

• numbers의 길이는 2 이상 100 이하입니다.
• numbers의 모든 수는 0 이상 100 이하입니다.

---

💡 개념 설명

• 중복 제거 및 정렬 준비: 합산 결과에서 중복된 값을 없애고 자동으로 오름차순 정렬하기 위해 **TreeSet<Integer>**를 생성합니다.
• 2중 반복문으로 조합 생성: 두 개의 for 루프를 사용하여 서로 다른 인덱스에 있는 두 수를 선택합니다. (바깥 루프 i는 0부터, 안쪽 루프 j는 **i + 1**부터 시작하여 중복 선택 방지)
• 합계 저장: 선택한 두 수의 합(numbers[i] + numbers[j])을 set.add() 메서드를 통해 저장합니다. 이때 TreeSet의 특성상 중복은 자동으로 제거되고 정렬됩니다.
• 배열 변환 및 반환: set에 저장된 데이터들을 결과 타입인 int[] 배열로 변환하여 반환합니다. (루프를 돌리거나 stream API 활용)

---

📎 입출력 예시

 

입출력 예 설명

입출력 예 #1

• 2 = 1 + 1 입니다. (1이 numbers에 두 개 있습니다.)
• 3 = 2 + 1 입니다.
• 4 = 1 + 3 입니다.
• 5 = 1 + 4 = 2 + 3 입니다.
• 6 = 2 + 4 입니다.
• 7 = 3 + 4 입니다.
• 따라서 [2,3,4,5,6,7] 을 return 해야 합니다.

입출력 예 #2

• 2 = 0 + 2 입니다.
• 5 = 5 + 0 입니다.
• 7 = 0 + 7 = 5 + 2 입니다.
• 9 = 2 + 7 입니다.
• 12 = 5 + 7 입니다.
• 따라서 [2,5,7,9,12] 를 return 해야 합니다.

---

📎 코드

import java.util.Arrays;
import java.util.TreeSet;

public class Solution {
    public int[] solution(int[] numbers) {
        TreeSet<Integer> set = new TreeSet<>();

        for (int i = 0; i < numbers.length; i++) {
            for (int j = i + 1; j < numbers.length; j++) {
                set.add(numbers[i] + numbers[j]);
            }
        }

        // Java 8 이상이라면 아래와 같이 한 줄로 배열 변환도 가능합니다.
        return set.stream().mapToInt(Integer::intValue).toArray();
    }

    public static void main(String[] args) {
        Solution sol = new Solution();

        // 테스트 케이스 1
        int[] n1 = {2, 1, 3, 4, 1};
        System.out.println("결과 1: " + Arrays.toString(sol.solution(n1))); // [2, 3, 4, 5, 6, 7]

        // 테스트 케이스 2
        int[] n2 = {5, 0, 2, 7};
        System.out.println("결과 2: " + Arrays.toString(sol.solution(n2))); // [2, 5, 7, 9, 12]
    }
}

 

---

📎 결과 

---