[자바(JAVA)] 두 개 뽑아서 더하기 / HashSet, TreeSet

 

✅ 내 풀이

import java.util.HashSet;
import java.util.Set;
import java.util.Collections;
import java.util.List;
import java.util.ArrayList;
    

class Solution {
    public int[] solution(int[] numbers) {
        
        Set<Integer> answer = new HashSet<>();
        
        for(int i = 0; i < numbers.length; i++) {
            for(int j = i + 1; j < numbers.length; j++) {  
                answer.add(numbers[i] + numbers[j]);   
            }
        }
        
        List<Integer> sortedList = new ArrayList<>(answer);
        Collections.sort(sortedList);   
        
        int[] result = new int[sortedList.size()];
        for(int i = 0; i < sortedList.size(); i++) {
            result[i] = sortedList.get(i);
        }
        
        return result;
    }
}

 

 

다른풀이에서 TreeSet을 사용하면 정렬까지 바로 된다고 해서 다시 리팩토링을 해봤다.

 

✅ 리팩토링

import java.util.*;

class Solution {
    public int[] solution(int[] numbers) {
        TreeSet<Integer> answer = new TreeSet<>();
        
        for (int i = 0; i < numbers.length; i++) {
            for (int j = i + 1; j < numbers.length; j++) {
                answer.add(numbers[i] + numbers[j]);
            }
        }
        
        int[] result = new int[answer.size()];
        int index = 0;
        for (int num : answer) {
            result[index++] = num;
        }
        
        return result;
    }
}