정수 배열 numbers가 주어집니다. numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return 하도록 solution 함수를 완성해주세요.
for 문을 통해 모든 경우의 수를 더해주고 중복을 없애기 위해 Set 에 결과를 담아주었다.
순서를 보장받지 못하는 Set 의 특성을 고려해 List 로 변환해 정렬을 해주었다.
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> answer = new ArrayList<>(set);
Collections.sort(answer);
return answer.stream().mapToInt(i -> i).toArray();
}
}
첫번째 시도에서 성공했지만 ,
댓글을 보던중 TreeSet 을 사용하면 자동으로 오름차순 정렬까지 된다는 글을 보아서 적용해 보기 위해 로직을 수정해 봤다.
TreeSet 은 존재는 알고 있었지만 기능은 처음 알게되었는데 정말 유용하게 사용될 것 같다.
import java.util.*;
class Solution {
public int[] solution(int[] numbers) {
TreeSet<Integer> treeSet = new TreeSet<>();
for (int i = 0; i < numbers.length; i++)
for (int j = i + 1; j < numbers.length; j++)
treeSet.add(numbers[i] + numbers[j]);
return treeSet.stream().mapToInt(i -> i).toArray();
}
}