C++ STL Sets

집합은 C++ 표준 템플릿 라이브러리 중 하나이다.
집합(Sets)은 특정한 순서로 특정 원소를 저장하는 컨테이너(container)이다.

주로 사용되는 함수들은 아래와 같다.
선언은 다음과 같이 한다.

sets<int> s; // creates a set of integers

집합의 크기는 다음과 같이 구할 수 있다.

int length = s.size(); // gives the size of the set.

집합에 새로운 원소는 insert() 함수를 이용한다.

s.insert(x); // inserts an integer x into the set s

원소를 지울 때는 erase() 함수를 이용한다.

s.erase(val); // erases an integer val from the set s.

집합에서 원소를 찾는 방법은 다음과 같다.

set<int>::iterator itr = s.find(val); // gives the iterator to the element val if it is found otherwise returns s.end()

예제

https://www.hackerrank.com/challenges/cpp-sets/problem?h_r=next-challenge&h_v=zen&isFullScreen=true

#include <cmath>
#include <cstdio>
#include <vector>
#include <iostream>
#include <set>
#include <algorithm>
using namespace std;


int main() {
    /* Enter your code here. Read input from STDIN. Print output to STDOUT */   
    int Q; // the number of queries
    int y, x; // y is the type of the query and x is an integer
    set<int> s;
    cin >> Q;

    for(int i = 0; i < Q; i++) {
        cin >> y >> x;
        if(y == 1) {
            s.insert(x);
        }
        else if(y == 2) {
            s.erase(x);
        }
        else if(y == 3) {
            if(s.find(x) == s.end()) {
                cout << "No" << endl;
            }
            else {
                cout << "Yes" << endl;
            }
        }
    }

    return 0;
}