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

---

입출력 예

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

---

입출력 예 설명

 

입출력 예 #1

• 2 = 1 + 1 입니다. (1이 numbers에 두 개 있습니다.)
• 3 = 2 + 1 입니다.
• 4 = 1 + 3 입니다.
• 5 = 1 + 4 = 2 + 3 입니다.
• 6 = 2 + 4 입니다.
• 7 = 3 + 4 입니다.
• 따라서 [2,3,4,5,6,7] 을 return 해야 합니다.

 

입출력 예 #2

• 2 = 0 + 2 입니다.
• 5 = 5 + 0 입니다.
• 7 = 0 + 7 = 5 + 2 입니다.
• 9 = 2 + 7 입니다.
• 12 = 5 + 7 입니다.
• 따라서 [2,5,7,9,12] 를 return 해야 합니다.
•  

---

• 내 풀이

import java.util.Set;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Arrays;

class Solution {
    public int[] solution(int[] numbers) {
        int[] answer = {};
        
        // 중복인 값을 피하기 위해 Set
        Set<Integer> set = new HashSet<>();

        for(int i = 0; i<numbers.length; i++){
        // 본인의 숫자를 제외하고 더해야하기 때문에 +1씩 붙음
            for(int j = 1+i; j<numbers.length; j++){
                set.add(numbers[i] + numbers[j]);
            }
        }
        // 중복이 제거된 set의 개수로 초기화
        answer = new int[set.size()];

        int count=0;
        //set의 값을 빼기 위한 Iterator 사용
        Iterator it = set.iterator();
        while(it.hasNext()){
            // Iterator 타입이 Object로 되어 있어서 int로 형변환
            // Iterator<Integer> it = set.iterator(); 로 하면 형변환 필요 없음
            answer[count] = (int) it.next();
            count++;
        }
        // 오름차순으로 정렬
        Arrays.sort(answer);

        return answer;
    }
}

HashSet 말고 TreeSet을 이용했으면 sort를 굳이 안해도 됐었다.

 

 

• 다른 사람 풀이

import java.util.*;

class Solution {
    public int[] solution(int[] numbers) {
        int[] answer = {};

        ArrayList<Integer> list = new ArrayList<> ();

        for(int i=0; i<numbers.length-1; i++) {
            for(int j=i+1; j<numbers.length; j++) {
                int sum = numbers[i] + numbers[j];
                if(! list.contains(sum)) list.add(sum);
            }
        }

        Collections.sort(list);

        answer = new int[list.size()];

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


        return answer;
    }
}