본문 바로가기
알고리즘 풀이/프로그래머스

[level 1] 두 개 뽑아서 더하기 - 68644 (JAVA)

by cornsilk-tea 2023. 5. 7.

문제 요약

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


문제 분석

Set을 사용해 중복값을 제거하며 이중반복문을 사용해 두 수를 더해가며 set에 저장 후 정렬하면 된다.


코드

import java.util.*;

class Solution {
    public int[] solution(int[] numbers) {
        // 결과값을 저장할 answer 배열을 선언
        int[] answer;
        
        // 중복값을 제외하며 숫자의 합을 저장할 Set을 선언.
        Set<Integer> set = new HashSet<>();
        
        // numbers 배열의 원소들을 순회하며 가능한 모든 조합의 합을 구한다.
        for (int i = 0; i < numbers.length; i++) {
            // j를 i+1부터 시작하여 자기 자신과의 합을 제외.
            for (int j = i + 1; j < numbers.length; j++) {
                // numbers[i]와 numbers[j]의 합을 set에 추가.
                set.add(numbers[i] + numbers[j]);
            }
        }
        
        // Set의 모든 원소를 int 배열로 변환하고 정렬.
        // 스트림을 사용하여 mapToInt를 통해 Integer 객체를 int로 변환하고,
        // sorted()를 사용하여 오름차순 정렬한 후 toArray()를 통해 배열로 변환.
        answer = set.stream().mapToInt(Integer::intValue).sorted().toArray();
        return answer;
    }
}

정리

새로 배운 내용

Java 8의 Stream API를 사용하여 Set을 int[] 배열로 쉽게 변환하고 정렬할 수 있다.

개선할 점

없음

더 찾아볼만한 사항

Java의 Stream API에 대해 더 학습하여 다양한 데이터 변환과정을 익히면 좋을 것 같다.

이 문제에선 중복을 허용하는 자료구조인 Set을 사용했지만, 다른 문제에선 중복을 허용하는 List를 사용할 수도 있다.