STL容器set用法以及codeforces 685B

Posted 2020-11-22 wushengyang

以前没怎么用过set,然后挂训练赛的时候发现set的妙用,结合网上用法一边学一边写.

首先set是一种容器,可以跟其他STL容器一样用 set<int > s 来定义, 它包含在STL头文件#include<set>中. 

其内部是用红黑树来实现的, 一个是保证set里面是有序的, 默认的排序是从小到大排序, 而且set容器的元素都是唯一的, 如果里面的元素允许重复那就用 multiset 容器来实现.

1. set的常用操作

s.begin()     返回set容器的第一个元素的位置(第一个迭代器)

s.end()       返回set容器的最后一个元素的位置(最后一个迭代器)

s.clear()     清空set容器

s.empty()     判断set容器是否为空

s.insert()    插入一个元素

s.erase()     删除一个元素

s.size()      返回容器里面有多少个元素

s.max_size()  返回set容器可以包含的元素最大个数

实例:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
set<int> s;
int main(int argc, char * argv[]){
    s.clear();
    if(s.empty()) cout<<"容器是空的"<<endl;
    else cout<<"容器不是空的"<<endl;
       s.insert(3);
       s.insert(6);
       s.insert(1);
       cout<<"set的第一个值为:"<<*s.begin()<<endl;
       cout<<"set的最后一个值为:"<<*s.end()<<endl;
       cout<<"set的maxsize的值为:"<<s.max_size()<<endl;
    return 0;
}

运行结果:

 

 

 

2. set的count() 函数

count() 用来查找set中某个值出现的次数, 但是由于set不允许出现重复的元素, 所以在set中用 s.count() 只会返回0或者1, 可以用来判断需要查找的数在set中存不存在, 而在multiset中就可以统计一个数字的出现个数了.

实例:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
set<int> s1;
multiset<int> s2;
int main(int argc, char * argv[]){
       s1.insert(3);
       s1.insert(6);
       s1.insert(1);
       s1.insert(1);
       s2.insert(3);
       s2.insert(6);
       s2.insert(1);
       s2.insert(1);
       cout<<"set中1出现的次数:"<<s1.count(1)<<endl;
       cout<<"set中2出现的次数:"<<s1.count(2)<<endl;
       cout<<"multiset中1出现的次数:"<<s2.count(1)<<endl;
       cout<<"multiset中2出现的次数:"<<s2.count(2)<<endl;
    return 0;
}

运行结果:

 

 

3.删除元素

s.earse(iterator)    　　删除iterator指向的值

s.earse(a,b)          　　删除定位器a和b之间的值

s.earse(key_value)      删除键值key_value的值

实例:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
set<int> s;
int main(int argc, char * argv[]){
    set<int>::iterator it;
       set<int>::iterator a;
       set<int>::iterator b;
       for(int i=1;i<=10;i++){
           s.insert(i);
       }
       cout<<"删除之前set中的元素是:"<<endl;
       for(it=s.begin();it!=s.end();it++){
           cout<<*it<<" ";
       }
       cout<<endl;
       s.erase(s.begin()); //删除指向的值
       cout<<"第一次删除之后set中的元素是:"<<endl;
       for(it=s.begin();it!=s.end();it++){
           cout<<*it<<" ";
       }
       cout<<endl;
       a=s.begin();
       cout<<"a指向的值是:"<<*a<<endl;
       b=s.begin();
       b++,b++,b++;
       cout<<"b指向的值是:"<<*b<<endl;
       s.erase(a,b); //删除(a,b)之间的值,a和b都是迭代器
       cout<<"第二次删除之后set中的元素是:"<<endl;
       for(it=s.begin();it!=s.end();it++){
           cout<<*it<<" ";
       }
       cout<<endl;
       s.erase(9); //删除值
       cout<<"第三次删除之后set中的元素是:"<<endl;
       for(it=s.begin();it!=s.end();it++){
           cout<<*it<<" ";
       }
       cout<<endl;
    return 0;
}

运行结果:

 

可以发现用第二种方法删除的值是 从a到b, 这里是包含a指向的元素而不包含b指向的元素, 就是 [a,b). 同时set中的删除操作是不会进行错误检查的, 比如定位器是否合法等等,所以用的时候要注意.

 

4.查找元素

前面说过用count()可以进行元素的查找, 还有一个函数find()也可以进行元素查找

count()可以查找元素存不存在,而find()可以找到元素的位置, 但是如果没有找到就返回end().

实例:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
set<int> s;
int main(int argc, char * argv[]){
    set<int>::iterator it;
    for(int i=1;i<=10;i++){
        s.insert(i);
    }
    cout<<"容器中的元素有:"<<endl;
    for(it=s.begin();it!=s.end();it++){
        cout<<*it<<" ";
    }
    cout<<endl;
    int a;
    cout<<"请输入要查找的元素:"<<endl;
    while(cin>>a){
        if(s.find(a)==s.end()) cout<<"没有找到该元素"<<endl;
        else cout<<*s.find(a)<<endl;
    }
    return 0;
}

运行结果:

 

 

 5.　s.lower_bound(key_value)　　　　返回第一个大于等于key_value的定位器

　　s.upper_bound(key_value)　　　　返回第一个大于key_value的定位器

　　这两个函数都是用二分来实现查找的,所以比较快.