프로그래머스(Java) - 두 개 뽑아서 더하기

안녕하세요.

방프개입니다.

 

프로그래머스 - 두 개 뽑아서 더하기 문제 링크

코딩테스트 연습 - 두 개 뽑아서 더하기

 

 

# 문제 설명

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

---

제한사항

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

---

입출력 예

numbersresult

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

---

입출력 예 설명

입출력 예 #1

• 2 = 1 + 1 입니다. (1이 numbers에 두 개 있습니다.)
• 3 = 2 + 1 입니다.
• 4 = 1 + 3 입니다.
• 5 = 1 + 4 = 2 + 3 입니다.
• 6 = 2 + 4 입니다.
• 7 = 3 + 4 입니다.
• 따라서 [2,3,4,5,6,7] 을 return 해야 합니다.

입출력 예 #2

• 2 = 0 + 2 입니다.
• 5 = 5 + 0 입니다.
• 7 = 0 + 7 = 5 + 2 입니다.
• 9 = 2 + 7 입니다.
• 12 = 5 + 7 입니다.
• 따라서 [2,5,7,9,12] 를 return 해야 합니다.

 

---

 

 

2가지 방법으로 풀어보았습니다.

 

#1

import java.util.List;
import java.util.LinkedList;
import java.util.Arrays;

class Solution {
    public int[] solution(int[] numbers) {
        int[] answer = {};
        int length = numbers.length;
        List<Integer> list = new LinkedList<>();

        for (int i=0; i<length-1; i++) {
            for (int j=i+1; j<length; j++) {
                int num = numbers[i] + numbers[j];

                if (list.indexOf(num) < 0) {
                    list.add(num);   
                }
            }
        }

        answer = new int[list.size()];

        for(int i=0; i<answer.length; i++) {
            answer[i] = list.get(i);
        }

        Arrays.sort(answer);

        return answer;
    }
}

첫 번째로 list를 활용해 풀어보았습니다.

2중 for문에서 각 숫자를 모두 더하고 그 안의 if문에서 indexOf를 활용해 기존의 list에 해당 값이 없다면 add를 하는 방법입니다.

(indexOf는 입력받은 문자의 값이 있으면 해당 값의 index를 return 해주고 없다면 -1을 return 해줍니다.)

 

그리고 이후에 answer 배열에 list의 값을 넣고 Arrays.sort를 통해 정렬해줍니다.

(Arrays.sort() 메서드의 매개 값으로기본 타입 배열이나 String배열을 지정해주면 자동으로 오름차순 정렬이 됩니다.)

 

 

#2

import java.util.Set;
import java.util.TreeSet;
import java.util.Iterator;

class Solution {
    public int[] solution(int[] numbers) {
        int[] answer = {};
        int length = numbers.length;
        
        Set<Integer> set = new TreeSet<>();
        
        for (int i=0; i<length-1; i++) {
            for (int j=i+1; j<length; j++) {
                set.add(numbers[i] + numbers[j]);
            }
        }
        
        answer = new int[set.size()];
        
        Iterator<Integer> iter = set.iterator();
        int i=0;
        
        while (iter.hasNext()) {
            answer[i++] = iter.next();
        }
        
        return answer;
    }
}

두 번째로는 set을 활용해 풀어보았습니다.

set을 활용한 이유는 answer 배열에 들어가는 합의 값들이 중복되어 들어가면 안 되기 때문인데요.

list는 입력된 순서를 유지하고 중복이 허용되는 반면 set은 입력의 순서가 없고 중복도 허용되지 않습니다.

 

set 중에서도 TreeSet을 활용해보았는데요. TreeSet의 경우에는 add 될 때 자동으로 오름차순으로 정렬이 되어 입력이 됩니다.

따라서 HashSet으로 입력을 해 다시 오름차순으로 정렬할 필요가 없어 TreeSet을 활용했습니다.

 

감사합니다.