[C++][STL] Sets-STL

https://www.hackerrank.com/challenges/cpp-sets/problem

 

 

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


int main() {
    set<int> s;
    set<int>::iterator itr;
    int num, q, val;
    
    cin >> num;
    
    for (int i = 0; i < num; i++) {
        cin >> q >> val;
        if (q == 1) s.insert(val);
        else if (q == 2) s.erase(val);
        else {
            itr = s.find(val);
            if (itr == s.end()) cout << "No" << endl;
            else cout << "Yes" << endl;
        }
    }
    return 0;
}

 

Reference

http://www.cplusplus.com/reference/set/set/