[JAVA] 프로그래머스 Lv.1 : 두 개 뽑아서 더하기

🔥 월간 코드 챌린지 시즌 1

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

 

 

예제:

Input:
numbers : [2, 1, 3, 4, 1]
Output:
[2,3,4,5,6,7]

Input:
numbers : [5,0,2,7]
Output:
[2,5,7,9,12]

 

제한 조건:

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

---

Solution #1

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

 

Result #1

 

💡 두 수를 더한 값을 Set 자료형에 저장시켜 중복을 없앤다. stream api를 이용하여 Set의 데이터를 정렬하고, int형 배열로 리턴시킨다.

---

Solution #2

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

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

        int[] answer = new int[set.size()];
        int i = 0;
        for (Integer integer : set) {
            answer[i] = integer;
            i++;
        }
        
        Arrays.sort(answer);
        
        return answer;
    }
}

 

Result #2

 

💡 stream api를 사용하지 않고, Iterator로 Set의 데이터를 int 배열에 저장하면 실행 시간이 굉장히 줄어든다.

 

---

 

More Algorithm!

 

👇👇

 

github.com/ggujangi/ggu.programmers

ggujangi/ggu.programmers

 

 

 

출처 : 프로그래머스