数据结构set的定义和常见引用

set

1 介绍

set容器中的元素不会重复，当插入集合中已有的元素时，并不会插入进去，而且set容器里的元素自动从小到大排序。

即：set里面的元素不重复 且有序

//头文件
#include<set>
//初始化定义
set<int> s;

Cpp

2 函数方法

代码	含义
s.begin()	返回set容器的第一个元素的地址（迭代器）O(1)
s.end()	返回set容器的最后一个元素的下一个地址（迭代器）O(1)
s.rbegin()	返回逆序迭代器，指向容器元素最后一个位置O(1)
s.rend()	返回逆序迭代器，指向容器第一个元素前面的位置O(1)
s.clear()	删除set容器中的所有的元素,返回unsigned int类型O(N)
s.empty()	判断set容器是否为空O(1)
s.insert()	插入一个元素
s.size()	返回当前set容器中的元素个数O(1)
erase(iterator)	删除定位器iterator指向的值
erase(first,second）	删除定位器first和second之间的值
erase(key_value)	删除键值key_value的值
查找
s.find(element)	查找set中的某一元素，有则返回该元素对应的迭代器，无则返回结束迭代器
s.count(element)	查找set中的元素出现的个数，由于set中元素唯一，此函数相当于查询element是否出现
s.lower_bound(k)	返回大于等于k的第一个元素的迭代器O(logN)
s.upper_bound(k)	返回大于k的第一个元素的迭代器O(logN)

3 访问

• 迭代器访问

for(set<int>::iterator it = s.begin(); it != s.end(); it++)
	cout << *it << " ";

Cpp

• 智能指针

for(auto i : s)
	cout << i << endl;

Cpp

• 访问最后一个元素

//第一种
cout << *s.rbegin() << endl;

Cpp

 //第二种
set<int>::iterator iter = s.end();
iter--;
cout << (*iter) << endl; //打印2;

Cpp

//第三种
cout << *(--s.end()) << endl;

Cpp

4 重载<运算符

• 基础数据类型

方式一：改变set排序规则，set中默认使用less比较器，即从小到大排序。（常用）

set<int> s1; // 默认从小到大排序
set<int, greater<int> > s2; // 从大到小排序

Cpp

方式二：重载运算符。（很麻烦，不太常用，没必要）

//重载 < 运算符
struct cmp {
    bool operator () (const int& u, const int& v) const {
       // return + 返回条件
       return u > v;
    }
};
set<int, cmp> s; 

for(int i = 1; i <= 10; i++)
    s.insert(i);
for(auto i : s)
    cout << i << " ";
// 10 9 8 7 6 5 4 3 2 1

Cpp

方式三：初始化时使用匿名函数定义比较规则

set<int, function<bool(int, int)>> s([&](int i, int j){
    return i > j; // 从大到小
});
for(int i = 1; i <= 10; i++)
    s.insert(i);
for(auto x : s)
    cout << x << " ";

Cpp

• 高级数据类型（结构体）

直接重载结构体运算符即可，让结构体可以比较。

struct Point {
	int x, y;
	bool operator < (const Point &p) const {
		// 按照点的横坐标从小到大排序,如果横坐标相同,纵坐标从小到大
		if(x == p.x)
			return y < p.y;
		return x < p.x;
	}
};

set<Point> s;
for(int i = 1; i <= 5; i++) {
    int x, y;
    cin >> x >> y;
    s.insert({x, y});
}	
/* 输入
5 4
5 2
3 7
3 5
4 8
*/

for(auto i : s)
    cout << i.x << " " << i.y << "\n";
/* 输出
3 5
3 7
4 8
5 2
5 4
*/

Cpp

5 其它set

multiset:元素可以重复，且元素有序

unordered_set ：元素无序且只能出现一次

unordered_multiset ： 元素无序可以出现多次

例题

B-关鸡_2024牛客寒假算法基础集训营1 (nowcoder.com)

#include<bits/stdc++.h>
using namespace std;
void slove(int n)
{
    set<pair<int,int >> s;
    int i,r=2,l=2;//r,l分别表示左右两边需要着火点的个数，各边2个，初始为2
        for(i=1;i<=n;i++)
        {
            int x,y;
            cin>>x>>y;
            s.insert({x,y});
            if(y<=0)l=1;//如果已知有着火点，则有着火点的一侧还需要1个着火点
            if(y>=0)r=1;//特殊的，如果为（2,0），此时左右需要的着火点都为1，所以只能用ifif语句
}
        for(auto[a1,a2]:s)//遍历集合s
        {
            for(i=-1;i<=1;i++)//查找是否有与已知着火点相对的3个位置其一是否有着火点
            {
                if(s.count({3-a1,a2+i}))
                {
                     if(a2<0)l=0;
                      if(a2>0)r=0;
                }
             }
         }
    int ans=3-s.count({2,0})-s.count({1,-1})-s.count({1,1});//与鸡紧挨的三个特殊位置（2,0）（1，-1）（1,1）
    ans=min(ans,l+r);
    cout<<ans<<endl;
}
int main()
{
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);//提高输入输出效率，缩短输入输出时间
    int t;
    cin>>t;
    while(t--)
    {
        int n;
        cin>>n;
        slove(n);
}
    return 0;
}