[JAVA] 두 개 뽑아서 더하기

 

 

문제

정수 배열 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]

 

작성 코드 및 풀이

import java.util.*;

class Solution {
    public int[] solution(int[] numbers) {
        int[] answer = {};
        ArrayList<Integer> list = new ArrayList<>();  // 동적 배열 생성
        
        for (int i=0; i<numbers.length-1; i++) {
            for (int j=i+1; j<numbers.length; j++) {
                int sum = numbers[i] + numbers[j];
                
                if (!list.contains(sum)) {            // list에 두 수의 합이 존재하지 않으면
                    list.add(sum);		      // 두 수의 합인 sum 추가
                }
            }
        }
        
        Collections.sort(list);                       // 오름차순 정렬
        
        answer = new int[list.size()];                // 두 수의 합을 저장할 배열 
        
        int idx = 0;
        
        for (int sum : list) {
            answer[idx] = sum;                        // list에서 하나씩 꺼내 answer 변수에 저장
            idx++;
        }
        
        return answer;
    }
}

 

다음과 같은 예시 데이터가 있다고 생각해보자.

 

5

1

3

2

6

 

1) 첫 번째 반복문에서는 큰 루프를 돌게 된다!

     즉, numbers[0] = 5일 때

           numbers[1] = 1일 때

                           ...

     이런식으로 비교를 하게 됨

 

2) 두 번째 반복문에서는 작은 루프를 돌게 된다!

     즉, numbers[0] = 5일 때 numbers[1], numbers[2], numbers[3], numbers[4]와 비교

           numbers[1] = 1일 때 numbers[2], numbers[3], numbers[4]와 비교

                                               ...

     이런식으로 반복문을 돌면서 각 배열의 합을 sum 변수에 저장해줌!

 

3) 이 때 해당 값이 list에 존재하지 않는다면 list 배열에 추가 (중복을 방지하기 위해)

 

4) 오름차순으로 정렬해주고, 향상된 for문을 이용하여 결과값을 리턴할 answer 배열에 값 저장!

 

* 향상된 for문에 대한 설명

2021.03.29 - [IT/개념정리] - 향상된 FOR 문

향상된 FOR 문

 

실행 결과