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

📚 두 개 뽑아서 더하기

[JAVA] 프로그래머스 Level 1.

코딩테스트 연습 - 두 개 뽑아서 더하기 | 프로그래머스 스쿨 (programmers.co.kr)

문제

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

제한사항

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

입출력 예

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

문제 해결 방안

정수 배열 numbers에 있는 두 개의 수를 뽑아서 더한 후 저장한다.

이 때 저장되는 값이 중복인 상황이 생기므로 자료구조 set을 이용하여 해결한다.

소스 코드

import java.util.*;
class Solution {
    public int[] solution(int[] numbers) {
        int[] answer = {};
        Set<Integer> set = new HashSet<>();
        
        for(int i=0; i<numbers.length-1; i++) {
            for(int j=i+1; j<numbers.length; j++) {
                set.add(numbers[i] + numbers[j]);
            }
        }

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

리팩토링 6.16

import java.util.*;
class Solution {
    public int[] solution(int[] numbers) {
        int[] answer = {};
        Set<Integer> set = new HashSet<>();
        
        for(int i=0; i<numbers.length-1; i++) {
            for(int j=i+1; j<numbers.length; j++) {
                set.add(numbers[i] + numbers[j]);
            }
        }        
        answer = new int[set.size()];
        int i=0;
        for(int num : set) {
            answer[i++] = num;
        }
        Arrays.sort(answer);
        return answer;
    }
}

stream().mapToInt().toArray()를 사용하면 간편하고 좋지만 단순한 반복문을 이용 후 Arrays.sort()가 속도면에서 빠르다.