두 개 뽑아서 더하기 (C++)

https://programmers.co.kr/learn/courses/30/lessons/68644

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

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> numbers) {
    vector<int> answer;
    int tmp_ans;

    for (int i = 0; i < numbers.size()-1; i++) {
        for (int j = i+1; j < numbers.size(); j++) {
            tmp_ans = numbers[i] + numbers[j];
            answer.push_back(tmp_ans);
        }
    }
    sort(answer.begin(), answer.end());
    answer.erase(unique(answer.begin(), answer.end()), answer.end());

    return answer;
}

 

이중 for 문을 사용하여 벡터 안에 두 값을 차례대로 더한 후

벡터 answer에 넣어주었다.

 

예를 들어 numbers가 [2, 1, 3, 4, 1]이라면

(2, 1) (2, 3) (2, 4) (2, 1),

(1, 3) (1, 4) . . . (4, 1) 이런 식으로 더해주었다.

 

그러면 벡터 answer 안에는 합들이 다 들어있기 때문에

오름차순 정렬을 위해 sort를 사용하여 이를 정렬해 주고,

unique를 사용하여 중복되는 원소들을 뒷부분으로 몰아준 후

erase를 사용하여 중복 원소들을 벡터에서 삭제해 주었다.