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

 

 

문제. 정수 배열 numbers가 주어집니다. numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return 하도록 solution 함수를 완성해주세요. 제한 사항. * numbers의 길이는 2 이상 100 이하입니다. * numbers의 모든 수는 0 이상 100 이하입니다.

 

 

나의 접근.

1. 모든 경우에 대해 합을 구한다.

2. TreeSet를 활용하여, 중복제거와 오름차순으로 정렬한다.

3. TreeSet를 int 배열로 가공하여 리턴한다.

 

import java.util.Arrays;
import java.util.TreeSet;

class Solution {
    public int[] solution(int[] numbers) {
        TreeSet<Integer> set = new TreeSet<>();

        for(int i = 0 ; i < numbers.length -1; i++){
            for(int j = i+1 ; j < numbers.length; j++){
                set.add(numbers[i] + numbers[j]);
            }
        }

        Integer[] ans = set.toArray(new Integer[0]);
        int[] answer = Arrays.stream(ans).mapToInt(i -> i).toArray();

        return answer;
    }
}

 

처음에는 위와 같이 TreeSet를 Integer 배열로 변환하고 Integer 배열를 int 배열로 변환하였다.

검색해보니, Set를 int배열로 변환하는 방법이 있어 다시 작성해보았다.

 

import java.util.Arrays;
import java.util.TreeSet;

class Solution {
	public int[] solution(int[] numbers) {
        TreeSet<Integer> set = new TreeSet<>();

        for(int i = 0 ; i < numbers.length -1; i++){
            for(int j = i+1 ; j < numbers.length; j++){
                set.add(numbers[i] + numbers[j]);
            }
        }
        
        int[] answer = set.stream().mapToInt(Integer::intValue).toArray();

        return answer;
    }
}

// 입력값
[2,1,3,4,1]

// 결과
[2,3,4,5,6,7]

 

 

 

링크.

https://school.programmers.co.kr/learn/courses/30/lessons/68644

https://bluechanyeong.tistory.com/entry/int-to-Integer-Integer-to-int