C++ :: STL set 컨테이너

set 컨테이너는 연결 컨테이너 중 가장 단순합니다. key라 불리는 원소(value)의 집합으로 이뤄진 컨테이너입니다. 균형 이진 트리로 구현되어 있습니다. 

 

출처: https://pangtrue.tistory.com/37

set은 모든 원소가 유일합니다. set은 특정 정렬 기준(조건자)에 따라 원소를 자동 정렬합니다. 다음은 원소를 set 컨테이너에 저장하는 예제입니다.

 

⭐️ set의 insert()

#include <iostream>
#include <set>
using namespace std;
 
int main(void){
    set<int> s;
 
    s.insert(50);
    s.insert(30);
    s.insert(80);
    s.insert(40);
    s.insert(10);
    s.insert(70);
    s.insert(90);
 
    set<int> :: iterator iter;
    for(iter = s.begin() ; iter != s.end() ; ++iter)
        cout << *iter << " ";
    cout << endl;
 
    s.insert(50);   // 중복된 원소는 저장되지 않는다
 
    for(iter = s.begin() ; iter != s.end() ; ++ iter)
        cout << *iter << " ";
    cout << endl;
 
    return 0;
}

 

⭐️ set의 insert()

10 30 40 50 70 80 90 
10 30 40 50 70 80 90 

 

다음 예제는 insert()의 멤버 함수 실패 시 반환값에 대한 예제입니다.

 

⭐️ insert()의 반환값(pair 객체)

#include <iostream>
#include <set>
using namespace std;
 
int main(void){
    set<int> s;
 
    pair<set<int> :: iterator, bool> pr;
 
    pr = s.insert(50);
    s.insert(40);
    s.insert(80);
 
    if(true == pr.second)
        cout << *pr.first << " 삽입 성공!" << endl;
    else
        cout << *pr.first << "가 이미 있씁니다. 삽입 실패!" << endl;
 
    set<int> :: iterator iter;
    for(iter = s.begin() ; iter != s.end() ; ++ iter)
        cout << *iter << " ";
    cout << endl;
 
    pr = s.insert(50);
 
    if(true == pr.second)
        cout << *pr.first << " 삽입 성공!" << endl;
    else
        cout << *pr.first << "가 이미 있습니다. 삽입 실패!" << endl;
 
    for(iter = s.begin() ; iter != s.end() ; ++ iter)
        cout << *iter <<  " ";
    cout << endl;
 
    return 0;
}

 

⭐️ insert()의 반환값(pair 객체) 실행결과

50 삽입 성공!
40 50 80 
50가 이미 있습니다. 삽입 실패!
40 50 80 

 

다음 에제는 반복자로 탐색 시작 위치를 지정함으로써 빠르게 원소를 삽입하는 insert() 예제입니다.

 

⭐️ set의 탐색 위치 지정 insert()

#include <iostream>
#include <set>
using namespace std;
 
int main(void){
    set<int> s;
    pair<set<int> :: iterator, bool> pr;
 
    s.insert(50);
    s.insert(30);
    s.insert(80);
    s.insert(40);
    s.insert(10);
    s.insert(70);
 
    pr = s.insert(90);          // pr.first는 90원소의 반복자
 
    set<int> :: iterator iter;
    for(iter = s.begin() ; iter != s.end() ; ++iter)
        cout << *iter << " ";
    cout << endl;
 
    s.insert(pr.first, 85);     // 90 원소의 반복자에서 검색 시작 후 삽입
 
    for(iter = s.begin() ; iter != s.end() ; ++iter)
        cout << *iter << " ";
    cout << endl;
 
    return 0;
}

 

⭐️ set의 탐색 위치 지정 insert() 실행결과

10 30 40 50 70 80 90 
10 30 40 50 70 80 85 90 

 

원소의 개수를 반환하는 count() 멤버 함수 예제입니다.

 

⭐️ set의 count()

#include <iostream>
#include <set>
using namespace std;
 
int main(void){
    set<int> s;
 
    s.insert(50);
    s.insert(30);
    s.insert(80);
    s.insert(40);
    s.insert(10);
    s.insert(70);
    s.insert(90);
 
    set<int> :: iterator iter;
    for(iter = s.begin() ; iter != s.end() ; ++iter)
        cout << *iter << " ";
    cout << endl;
 
    cout << "원소 50개 개수: " << s.count(50) << endl;
    cout << "원소 100개 개수: " << s.count(100) << endl;
 
    return 0;
}

 

⭐️ set의 count() 실행결과

10 30 40 50 70 80 90 
원소 50개 개수: 1
원소 100개 개수: 0

 

다음은 원소를 검색하는 finc() 멤버 함수 예제입니다.

 

⭐️ set의 find()

#include <iostream>
#include <set>
using namespace std;
 
int main(void){
    set<int> s;
    s.insert(50);
    s.insert(30);
    s.insert(80);
    s.insert(40);
    s.insert(10);
    s.insert(70);
    s.insert(90);
 
    set<int> :: iterator iter;
    for(iter = s.begin() ; iter != s.end() ; ++iter)
        cout << *iter << " ";
    cout << endl;
 
    iter = s.find(30);
    if(iter != s.end())
        cout << *iter << "가 s에 있다!" << endl;
    else
        cout << "20이 s에 없다!" << endl;
 
    return 0;
}

 

⭐️ set의 find() 실행결과

10 30 40 50 70 80 90 
30가 s에 있다!

 

다음은 key(원소)의 구간을 반환하는 lower_bound() 와 upper_bound 예제입니다.

 

⭐️ set의 lower_bound(), upper_bound()

#include <iostream>
#include <set>
using namespace std;
 
int main(void){
    set<int> s;
    s.insert(50);
    s.insert(30);
    s.insert(80);
    s.insert(40);
    s.insert(10);
    s.insert(70);
    s.insert(90);
 
    set<int>::iterator iter;
    for(iter = s.begin() ; iter != s.end() ; ++iter)
        cout << *iter << " ";
    cout << endl;
 
    set<int>::iterator iter_lower;
    set<int>::iterator iter_upper;
 
    iter_lower = s.lower_bound(30);
    iter_upper = s.upper_bound(30);
    cout << *iter_lower << endl;
    cout << *iter_upper << endl;
 
    iter_lower = s.upper_bound(55);
    iter_upper = s.upper_bound(55);
    if(iter_lower != iter_upper)
        cout << "55가 s에 있음!" << endl;
    else
        cout << "55가 s에 없음!" << endl;
 
    return 0;
}

 

⭐️ set의 lower_bound(), upper_bound() 실행결과

10 30 40 50 70 80 90 
30
40
55가 s에 없음!

 

📚 출처

공동환(2012), 뇌를 자극하는 C++ STL, 한빛미디어.