[프로그래머스 코딩테스트] 두 개 뽑아서 더하기(Java)

  문제 설명

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

 

  제한사항

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

 

  입출력 예

numbers	result
[2, 1, 3, 4, 1]	[2, 3, 4, 5, 6, 7]
[5, 0, 2, 7]	[2, 5, 7, 9, 12]

import java.util.HashSet;
import java.util.ArrayList;
import java.util.Collections;
class Solution {
    public int[] solution(int[] numbers) {
        HashSet<Integer> set = new HashSet<>();
        for (int i = 0; i < numbers.length; i++) {
            for (int j = 0; j < numbers.length; j++) {
                if (i != j) {
                    set.add(numbers[i] + numbers[j]);
                }
            }
       }
        ArrayList<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;
    }
}

  이 문제는 중복값을 허용하지 않으므로 HashSet을 이용하여 중복값이 배열에 추가되지 않도록 합니다. for문이 0부터 numbers의 길이만큼 순회합니다. i와 j는 각각 for문을 순회하면서 i와 j가 같지 않을 때 값을 더하여, set에 데이터가 추가됩니다. 각각 0부터 시작이 아닌, i는 0부터 numbers.length - 1까지, j는 1부터 numbers.length까지 순회했다면 연산을 덜 하고 계산은 빠른, 효율적인 식이 되었을 것 같습니다.

  set 값을 가져오는 ArrayList를 생성합니다. Collections 클래스의 sort를 사용하여 오름차순으로 정렬합니다. 만약 HashSet이 아닌 TreeSet을 썼다면, 이 과정은 필요하지 않았을 것입니다. TreeSet은 기본적으로 오름차순으로 값이 정렬되기 때문입니다.

  연산은 끝났지만, 배열을 하나 더 만든 이유는 반환이 제대로 되지 않아서입니다. array타입만 정답으로 인정이 되는 듯해서, for문으로 ArrayList를 다시 배열로 옮겨주었습니다. HashSet을 배열로 바로 옮겨오는 법을 몰라서 상당히 번거로운 풀이가 되었네요.