[프로그래머스] 두 개 뽑아서 더하기

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

풀이

numbers의 길이가 100 이하이므로 완전 탐색으로 해결할 수 있다. 숫자의 중복이 있을 수 있으므로 Set을 사용해서 해결했다.

코드

C++

#include <string>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;

vector<int> solution(vector<int> numbers) {
    set<int> s;
    for(int i=0; i<numbers.size()-1; i++)
        for(int j=i+1; j<numbers.size(); j++)
            s.insert(numbers[i] + numbers[j]);

    vector<int> answer;
    for(int sum : s)
        answer.push_back(sum);
    sort(answer.begin(), answer.end());
    return answer;
}

Python3

def solution(numbers):
    result = set()
    n = len(numbers)
    for i in range(n - 1):
        for j in range(i + 1, n):
            result.add(numbers[i] + numbers[j])
    answer = list(result)
    answer.sort()
    return answer