学会STL切遍天下水题

set

set用二叉搜索树实现，集合中的每个元素只出现一次，并且是排好序的。
访问元素的时间复杂度是O(logn)，非常高效。

模板题

Galactic Collegiate Programming Contest

#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define U unsigned
#define P std::pair<int,int>
#define LL long long
#define pb push_back
#define MP std::make_pair
#define V std::vector<int>
#define all(x) x.begin(),x.end()
#define CLR(i,a) memset(i,a,sizeof(i))
#define FOR(i,a,b) for(int i = a;i <= b;++i)
#define ROF(i,a,b) for(int i = a;i >= b;--i)
#define DEBUG(x) std::cerr << #x << '=' << x << std::endl
#define INF 0x3f3f3f3f
const int N=1e5+5;

struct _node{
	int id,a,b;
	bool operator < (const _node k) const{
		if(a==k.a&&b==k.b) return id>k.id;   //相等的话优先把当前放后面
		else return a==k.a?b>k.b:a<k.a;  //题目做得越少越塞前面 
	}
	_node(int id,int a,int b):id(id),a(a),b(b){}
};

set<_node> s;
int a[N],b[N]; 

int main(){
	int n,m;scanf("%d%d",&n,&m);
	FOR(i,1,m){
		int x,y;scanf("%d%d",&x,&y);
		if(x==1){
			a[1]++,b[1]+=y;
			if(!s.empty()){
				//把第一个小于他的数字之后的都干掉。
				//因为我们是要找lower_bound，是找大的，所以在set里面
				//我们反着排序，越垃圾的越放前面。 
				set<_node>::iterator it=s.lower_bound(_node(1,a[1],b[1]));
				if(s.begin()!=it) s.erase(s.begin(),it); //把前面都干掉 
			}
		}else{
			if(s.find(_node(x,a[x],b[x]))==s.end()){
				a[x]++,b[x]+=y;
				if(a[x]>a[1]||(a[x]==a[1]&&b[x]<b[1]))
					s.insert(_node(x,a[x],b[x]));
			}else{
				s.erase(_node(x,a[x],b[x]));
				a[x]++,b[x]+=y;
				s.insert(_node(x,a[x],b[x]));
			}
		}
		printf("%d\n",s.size()+1);
	}
}