프로그래머스(Programmers)

두 개 뽑아서 더하기

cosmohoo 2022. 2. 16. 23:00
반응형

문제 설명

 

 

입출력 예

numbers                                                                                 result

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

입출력 예 설명

입출력 예 #1

  • 2 = 1 + 1 입니다. (1이 numbers에 두 개 있습니다.)
  • 3 = 2 + 1 입니다.
  • 4 = 1 + 3 입니다.
  • 5 = 1 + 4 = 2 + 3 입니다.
  • 6 = 2 + 4 입니다.
  • 7 = 3 + 4 입니다.
  • 따라서 [2,3,4,5,6,7] 을 return 해야 합니다.

입출력 예 #2

  • 2 = 0 + 2 입니다.
  • 5 = 5 + 0 입니다.
  • 7 = 0 + 7 = 5 + 2 입니다.
  • 9 = 2 + 7 입니다.
  • 12 = 5 + 7 입니다.
  • 따라서 [2,5,7,9,12] 를 return 해야 합니다.

 

#include <string>
#include <vector>
#include <set>

using namespace std;

vector<int> solution(vector<int> numbers) {
    vector<int> answer;
    //set<int, greater<int>> numSet; 내림차순 표현법
    set<int, less<int>> numSet; //오름차순 표현법
    for(int i=0; i<numbers.size()-1; i++) // 첫번째 INDEX
    {
        for(int j=i+1; j<numbers.size(); j++) // 두번째 INDEX
        {
            int tmp = numbers[i] + numbers[j];
            numSet.insert(tmp);
        }
    }
    
    for(auto a : numSet) //정렬된 값 vector에 삽입
    {
        answer.push_back(a);
    }
    return answer;
}

 

=> C++에 있는 <set> 자료형을 사용할 수 있으면 간단히 풀 수 있는 문제였습니다. 

=> set은 key값으로 이루어져있으며, 중복을 허용하지 않습니다. 

=> 위 사항을 염두에 두고 문제에 주어진대로 코딩하였습니다. 

=> 풀이는 아래의 그림으로 참조하겠습니다. 

 

문제풀이

 

반응형