문제
접근
배열의 원소 중 두개를 뽑아 더한 결과의 집합을 구하는 문제이다.
완전탐색을 해주면 되는데, 중복되는 결과를 어떻게 제거하는지가 관건이다.
파이썬의 경우 set() 을 이용하여 쉽게 해결이 가능하다.
C++의 경우는 STL을 사용하면 set을 이용하여 쉽게 해결이 가능하다.
이 문제의 경우 주어지는 numbers의 크기를 보면, 모든 원소는 0~100사이의 수이다.
따라서 이 중 두 개를 뽑아서 더해도 결과는 \(0 \leqq result \leqq 200\) 이므로,
그냥 해시를 이용하여 카운팅해주어도 된다.
아마 속도는 해시를 사용하는게 가장 빠르지 않을까?
코드
- C++ 코드 (STL 사용)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#include <string>
#include <vector>
#include <set>
using namespace std;
vector<int> solution(vector<int> numbers) {
set<int> s;
for(int i = 0; i<numbers.size(); i++){
for(int j = i+1; j<numbers.size(); j++){
s.insert(numbers[i] + numbers[j]);
}
}
vector<int> answer(s.begin(), s.end());
return answer;
}
- C++ 코드 (STL 미사용)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#include <vector>
using namespace std;
vector<int> solution(vector<int> numbers) {
int arr[205] = {0};
vector<int> answer;
for(int i = 0; i<numbers.size(); i++){
for(int j = i+1; j<numbers.size(); j++){
arr[numbers[i] + numbers[j]]++;
}
}
for(int i = 0; i<201; i++){
if(arr[i] > 0)
answer.push_back(i);
}
return answer;
}
- 파이썬 코드
1
2
3
4
5
6
def solution(numbers):
answer = set()
for i in range(len(numbers)):
for j in range(i+1, len(numbers)):
answer.add(numbers[i]+numbers[j])
return sorted(list(answer))