C++ STL 学习| set

C++ STL 学习| set

set

set的定义

set翻译为集合，是一个内部自动有序且不含重复元素的容器，当遇到需要去掉重复元素的情况且因元素较大或非int型不能直接用散列表，考虑使用set来保留元素本身而不考虑个数。

如需使用set，加头文件#include<set>，并加上一句using namespace std;即可

单独定义一个set

set<typename> name;

typename可为任何基本类型，如int、double、char、结构体等，也可以是STL标准容器，如vector，set、queue等。

set<int> name;
set<double> name;
set<char> name;
set<node> name;  //node为结构体类型

set<set<int> > name;  //定义二维数组

定义set数组

set<typename> Arrayname[arraySize];  //定义set数组
//如：
set<int> vi[100];

Arrayname[0]~Arrayname[arraySize - 1]中的每一个都是set容器。

set容器内元素的访问

set无法通过下标进行访问，只能通过迭代器(iterator)访问。

只能通过迭代器访问

迭代器（iterator）可以理解为一种类似于指针的东西，其定义是：

set<typename>::iterator it;

这样it就是一个set<typename>::iterators型的变量，如

set<int>::iterator it;
set<double>::iterator it;

这样就得到了迭代器it，并且可以通过*it来访问vector里的元素。

但是除了vector和string之外的STL容器都不支持*(it + i)的访问方式，只能使用枚举的方式。

#include<stdio.h>
#inlcude<set>
using namespace std;
int main() {
    set<int> st;
    st.insert(3);  //insert(x)将x插入set中
    st.insert(5);
    st.insert(2);
    st.insert(3);
    for(set<int>::iterator it = st.begin(); it != st.end(); i++) {  //注：此处不支持it < st.end()的写法
        printf("%d ", *it);
    }
    return 0;
}  //其输出为：2 3 5

可以发现，set内的元素自动递增排序且自动去除了重复元素。再次注意：在常用STL容器中，只有在vector和string中才允许使用vi.begin() + i这种迭代器加上整数的写法！

set常用函数实例解析

（1）insert()

insert(x)可将x插入set容器中，并自动递增排序和去重，时间复杂度为O(logN)，N为set内元素个数。示例可见上处代码，注意与vector中的insert()函数相区别！

（2）find()

find(value)返回set中对应值为value的迭代器，时间复杂度为O(logN)，N为set内元素个数。

#include<stdio.h>
#include<set>
using namespace std;
int main() {
    set<int> st;
    for(int i = 1; i <= 3; i++) {
       st.insert(i);
    }
    set<int>::iterator it = st.find(2);
    printf("%d", *it);
    //也可以将上两句代码合并为
    //printf("%d", *(st.find(2)));
    return 0;
}

（3）erase()

erase()有两种用法：删除单个元素、删除一个区间内的所有元素。

①删除单个元素

删除单个元素有两种方法：

• st.erase(it)，即删除迭代器为it处的元素，时间复杂度为O(1)，可以结合find()函数使用。

#include<stdio.h>
#include<set>
using namespace std;
int main() {
    set<int> st;
    st.insert(100);
    st.insert(200);
    st.insert(100);
    st.insert(300);
    //插入后的st为100 200 300
    st.erase(st.find(100));
    st.erase(st.find(200));
    //找到并删除100和200
    for(set<int>::iterator it = st.begin(); it != st.end(); it++) {
        printf("%d ",*it);
    }
    return 0;
}

• st.erase(value)，value为所需删除元素的值，时间复杂度为O(logN)，N为set内元素个数

#include<stdio.h>
#include<set>
using namespace std;
int main() {
    set<int> st;
    st.insert(100);
    st.insert(200);
    st.erase(100);  //删除st值为100的元素
    for(set<int>::iterator it = st.begin(); it != st.end(); it++) {
        printf("%d ", *it);
    }
    return 0;
}

②删除一个区间内的所有元素

erase(first, last)即删除区间[first, last)内的所有元素，时间复杂度为O(last - first)

#include<stdio.h>
#include<set>
using namespace std;
int main() {
    set<int> st;
    st.insert(20);
    st.insert(10);
    st.insert(40);
    st.insert(30);
    //使用find找到删除区间的左值
    set<int>::iterator it = st.find(30);
    //删除区间[it, end())内的元素
    st.erase(it, st.end());
    for(it = st.begin(); it != st.end(); it++) {
        printf("%d ", *it);
    }
    //输出10 20
    return 0;
}

（4）size()

size()用来获得set中元素的个数，时间复杂度为O(1)。size()返回的是unsigned类型，一般用%d也没什么问题，这一点对所有STL容器都是一样的

#include<stdio.h>
#include<set>
using namespace std;
int main() {
    set<int> st;
    st.insert(10);
    st.insert(40);
    st.insert(20);
    st.insert(50);
    printf("%d", st.size());
    return 0;
}

（5）clear()

clear()用来清空set中的所有元素，时间复杂度为O(logN)，N为set中元素个数。

#include<stdio.h>
#include<set>
using namespace std;
int main() {
    set<int> st;
    st.insert(40);
    st.insert(20);
    st.insert(30);
    st.insert(10);
    st.clear();
    printf("%d", st.size());
    return 0;
}

总结：set最主要的作用是自动去重并按升序排序，因此碰到需要去重但是不方便直接开数组的情况，可以考虑使用set解决。set中元素具有唯一性，如果需要处理不唯一情况，需要使用multiset，且C++11增加了unordered_set，以散列代替set内部的红黑树实现，使其可以用来处理只去重但不排序的需求，速度比set快得多。