[programmers] 두 개 뽑아서 더하기 (java) -1level

2022. 3. 1. 22:25Algorithm/Programmers

문제 설명

 

 

문제 해결

 

Set Collection 중에서 중복 없는 HashSet을 이용한다.

서로 다른 인덱스의 값을 더해주고 배열로 가져와 오름차순으로 정렬해주었더니 문제가 풀렸다.

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

public class 두개뽑아서더하기 {
    public int[] solution(int[] numbers) {
        Set<Integer> set = new HashSet<Integer>();

        for (int i = 0; i < numbers.length; i++) {
            for (int j = i + 1; j < numbers.length; j++) {
                set.add(numbers[i] + numbers[j]);
            }
        }
        ArrayList<Integer> list = new ArrayList<>(set);
        int[] sum = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            sum[i] = list.get(i);
        }

        Arrays.sort(sum);
        return sum;
    }
}

 

+ ) 다른 사람의 풀이

 

stream() 을 이용할 경우 코드는 간결해지지만 속도측면에서 느릴 수 있다. 명심!

import java.util.HashSet;
import java.util.Set;

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]);
            }
        }

        return set.stream().sorted().mapToInt(Integer::intValue).toArray();
    }
}
728x90