[Java] Lv1. 두 개 뽑아서 더하기

🔗 문제 링크

프로그래머스

 

📝 문제 설명

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

⚠️ 제한 사항

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

 

✨ 내 코드

import java.util.ArrayList;
import java.util.Arrays;

class Solution {
    public int[] solution(int[] numbers) {
        ArrayList<Integer> newArr = new ArrayList<>();
        
        for (int i = 0; i < numbers.length - 1; i++) {
            for (int j = i + 1; j < numbers.length; j++) {
                newArr.add(numbers[i] + numbers[j]);
            }
        }
        
        return newArr.stream().distinct().sorted().mapToInt(Integer::intValue).toArray();
    }
}

 

💡 풀이 및 회고

주어진 배열과 숫자의 크기가 충분히 작기 때문에 반복문을 여러번 돌려도 된다고 판단했다.

중첩 for문을 통해 모든 숫자를 한번씩 서로 더해줬다.

이제 정렬 후 중복을 제거하기만 하면 되는데 반복문 돌려서 하나씩 비교하는 것 말고는 마땅한 방법이 생각이 나지 않았다.

결국 중복 제거는 책의 풀이를 따라했다.

중복제거를 이렇게 코드 한 줄로 처리할 수 있다니 외워둬야겠다.

중복제거 코드 흐름

`newArr.stream().distinct().sorted().mapToInt(Integer::intValue).toArray();`

ArrayList 자체에는 중복 제거 메소드가 없기 때문에 Stream으로 바꿔준다.

이때 주의해야할 점은 Stream은 Collection이며 Collection은 참조형을 전제로 한 메소드라는 점이다.

이 때문에 newArr를 Arrays가 아닌 ArrayList로 선언했다.

문제의 리턴 타입이 int[]이므로 Integer를 int로 매핑하고 toArray() 메소드로 바꿔준다.

상황에 따라서 mapToInt, toArray는 생략할 수 있을 것 같다.