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

 

두 개 뽑아서 더하기

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

 

 

입출력 예

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

 

 

2중 반복문을 이용하고 vector의  find 연산을 이용하여 중복 값을 체크하고 마지막에 정렬하여 풀이하였습니다.

 

 

 

풀이

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

using namespace std;

vector<int> solution(vector<int> numbers) {
    vector<int> answer;
    for(int i = 0; i < numbers.size() - 1; i++){
        for(int j = i + 1; j < numbers.size(); j++){
            int num = numbers[i] + numbers[j];
            if(find(answer.begin(), answer.end(), num) == answer.end()) answer.push_back(num);
        }
    }
    sort(answer.begin(), answer.end());
    return answer;
}

 

 

 

 

추가적으로 set STL을 활용하면 set의 중복을 허용하지 않는 특성을 가지고 풀이 할 수도 있습니다.

 

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

vector<int> solution(vector<int> numbers) {
    vector<int> answer;
    set<int> st;
    for(int i = 0;i<numbers.size();++i){
        for(int j = i+1 ; j< numbers.size();++j){
            st.insert(numbers[i] + numbers[j]);
        }
    }
    answer.assign(st.begin(), st.end());
    return answer;
}

// 박태수 , 비밀스럽게 , 백승호 , FFudi 외 5 명