[Java] HashSet, set → 배열 리턴 / 두 개 뽑아서 더하기

🔗 링크

두 개 뽑아서 더하기

프로그래머스

📌 문제 설명

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

 

👀 예제

✏️ 풀이 순서

1. 중복 제거를 위해 Set 사용
2. 두 수의 합 구하기
3. 배열로 변환해 리턴

📍 HashSet

• 중복제거: HashSet은 중복된 요소를 자동으로 제거함.
• 빠른 조회 속도: 내부적으로 해시 테이블 사용하므로 요소를 추가하거나 조회하는 작업이 평균적으로 O(1)의 시간 복잡도를 가짐

💻 코드

import java.util.*;

class Solution {
    public int[] solution(int[] numbers) {
        
        Set<Integer> sum = new HashSet<>();
        
        for (int i = 0; i < numbers.length; i++) {
            for (int j = i + 1; j < numbers.length; j++) {
                sum.add(numbers[i] + numbers[j]);
            }
        }
        
        List<Integer> list = new ArrayList<>(sum);
        Collections.sort(list);
        
        return list.stream().mapToInt(Integer::intValue).toArray();
    }
}

→ 굳이 list로 바꾸지 않고 set에서 바로 배열로 바꿀 수 있었다.

 

다른 사람의 코드

import java.util.*;

class Solution {
    public int[] solution(int[] numbers) {
        
        Set<Integer> sum = new HashSet<>();
        
        for (int i = 0; i < numbers.length; i++) {
            for (int j = i + 1; j < numbers.length; j++) {
                sum.add(numbers[i] + numbers[j]);
            }
        }
        
        return sum.stream().sorted().mapToInt(i->i).toArray();
    }
}

 

---

📚Reference