Algorithm/Programmers

[Programmers - C++] 두 개 뽑아서 더하기

tjddneva 2022. 2. 6. 19:18

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

 

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

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

programmers.co.kr

 

set 만 쓰면 굉장히 쉬운 문제이다! set 자료구조에서 insert 함수는 집어넣는 원소가 이미 있으면 집어넣지 않는 것으로 알고 있는데 이래서 set 만 쓰면 된다!

 

이중 for 문을 써서 배열을 돌면서 합을 set 에 넣고 answer vector 에 다시 옮겨 담아주면 끝이다.

 

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

vector<int> solution(vector<int> numbers) {
    
    vector<int> answer;
    set<int> temp;
    int sum;
    for(int i=0; i<numbers.size(); i++){
        for(int j=i+1; j<numbers.size(); j++){
            sum = numbers[i] + numbers[j];
            temp.insert(sum);
        }
    }
    
    for(auto c : temp){
        answer.push_back(c);
    }
    return answer;
}