알고리즘(Algorithm)/프로그래머스(Programmers)

[프로그래머스] 두 개 뽑아서 더하기 - JAVA

나아가는중 2021. 10. 24. 00:19
반응형

프로그래머스 두 개 뽑아서 더하기 - JAVA


문제 설명

  1. 정수 배열 numbers가 주어진다.
  2. numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return.

제한 사항

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

문제 풀이

  1. 배열을 2중 for문으로 순회하며 만들 수 있는 모든 수를 저장
    • 중복이 발생할 수 있으므로 HashSet을 사용하여 저장.
  2. 1에서 찾은 모든 수를 정렬.
    • HashSet은 정렬이 불가능하여 ArrayList로 복사하여 정렬.
  3. ArrayList의 수들을 배열에 담아 return.

소스 코드

import java.util.*;

class Solution {
    public int[] solution(int[] numbers) {        
        Set<Integer> set = new HashSet<>();
        for (int i = 0; i < numbers.length; i++) {
            for (int j = i + 1; j < numbers.length; j++) {
                set.add(numbers[i] + numbers[j]);
            }
        }

        List<Integer> list = new ArrayList<>(set);
        Collections.sort(list);

        int[] answer = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            answer[i] = list.get(i);
        }

        return answer;
    }
}
반응형