bzoj 1208: [HNOI2004]宠物收养所 stl

→题目链接←

【想说的话】

又是一道stl可以水过的好题

本来有输入输出优化的...但是为了缩代码，删掉了

【题解】

不用管是宠物还是人来了，如果另一方有剩余，那就直接找最接近的加进答案，否则把他放进这一方的set

咋找set搞一搞就好了，代码应该挺可看的-___-

【代码】

#include<bits/stdc++.h>

#define ll long long
#define mod 1000000

using namespace std;

int n;
ll ans=0;
set<ll>s[2];
set<ll>::iterator r,r1;

int main(){
	scanf("%d",&n);
	while(n--){
		int x;
		ll y;
		scanf("%d%lld",&x,&y);
		if(s[x^1].empty()){
			s[x].insert(y);
			continue;
		}
		r=s[x^1].lower_bound(y);
		if(r==s[x^1].end())r--;
		if(r!=s[x^1].begin()){
			r1=r,r1--;
			if(abs(*r1-y)<=abs(*r-y))r--;
		}
		ans+=abs(*r-y)%mod;
		ans%=mod;
		s[x^1].erase(r);
	}
	printf("%lld\n",ans);
	
	return 0;
}