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;
}