[Programmers] 두 개 뽑아서 더하기 - Java

문제 

프로그래머스

 

풀이

배열에 있는 2개의 수를 뽑은 합을 구해서 오름차순 정렬하는 문제이다.

ArrayList, 동적 배열을 사용해서 합들을 다 담아서 오름차순 정렬하려고 했으나, 수가 겹칠 수 있는 문제가 존재한다.

[1,1,1] 이라면[2,2,2]가 아니라 [2]가 되어야하기 때문이다.

 

그래서 정렬을 유지하면서 값을 중복하지 않게 담을 수 있는 자료구조인 TreeSet을 선택했다.

 

1. TreeSet으로 배열에서 다른 인덱스에서 뽑은 2개의 수를 더한 값을 담는다.

 

2. set의 크기만큼 배열을 만든다.

 

3. 배열에 set에 담겨있는 값을 넣는다.

 

소스코드

import java.util.TreeSet;

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]);
            }
        }
        
        int[] answer = new int[set.size()];
        
        int idx = 0;
        
        for (int v : set) {
            answer[idx++] = v;
        }
        
        return answer;
    }
}

 

실행결과