题目链接:
https://www.lydsy.com/JudgeOnline/problem.php?id=1208
题目大意:
最近,阿Q开了一间宠物收养所。收养所提供两种服务:收养被主人遗弃的宠物和让新的主人领养这些宠物。每个领养者都希望领养到自己满意的宠物,阿Q根据领养者的要求通过他自己发明的一个特殊的公式,得出该领养者希望领养的宠物的特点值a(a是一个正整数,a<2^31),而他也给每个处在收养所的宠物一个特点值。这样他就能够很方便的处理整个领养宠物的过程了,宠物收养所总是会有两种情况发生:被遗弃的宠物过多或者是想要收养宠物的人太多,而宠物太少。 1. 被遗弃的宠物过多时,假若到来一个领养者,这个领养者希望领养的宠物的特点值为a,那么它将会领养一只目前未被领养的宠物中特点值最接近a的一只宠物。(任何两只宠物的特点值都不可能是相同的,任何两个领养者的希望领养宠物的特点值也不可能是一样的)如果有两只满足要求的宠物,即存在两只宠物他们的特点值分别为a-b和a+b,那么领养者将会领养特点值为a-b的那只宠物。 2. 收养宠物的人过多,假若到来一只被收养的宠物,那么哪个领养者能够领养它呢?能够领养它的领养者,是那个希望被领养宠物的特点值最接近该宠物特点值的领养者,如果该宠物的特点值为a,存在两个领养者他们希望领养宠物的特点值分别为a-b和a+b,那么特点值为a-b的那个领养者将成功领养该宠物。 一个领养者领养了一个特点值为a的宠物,而它本身希望领养的宠物的特点值为b,那么这个领养者的不满意程度为abs(a-b)。【任务描述】你得到了一年当中,领养者和被收养宠物到来收养所的情况,希望你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时,收养所里面既没有宠物,也没有领养者。
思路:
直接用set即可。
注意特判迭代器,否则会RE
#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(false);//不可再使用scanf printf
#define Max(a, b) ((a) > (b) ? (a) : (b))//禁用于函数,会超时
#define Min(a, b) ((a) < (b) ? (a) : (b))
#define Mem(a) memset(a, 0, sizeof(a))
#define Dis(x, y, x1, y1) ((x - x1) * (x - x1) + (y - y1) * (y - y1))
#define MID(l, r) ((l) + ((r) - (l)) / 2)
#define lson ((o)<<1)
#define rson ((o)<<1|1)
#define Accepted 0
#pragma comment(linker, "/STACK:102400000,102400000")//栈外挂
using namespace std;
inline int read()
{
int x=,f=;char ch=getchar();
while (ch<''||ch>''){if (ch=='-') f=-;ch=getchar();}
while (ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
return x*f;
}
typedef long long ll;
const int maxn = + ;
const int MOD = ;//const引用更快,宏定义也更快
const int INF = 1e9 + ;
const double eps = 1e-;
const double pi = acos(-); set<int>s, s1;
set<int>::iterator l, r, cnt; int main()
{
int n, m, T;
scanf("%d", &n);
int x, flag;
int ans = ;
for(int i = ; i <= n; i++)
{
scanf("%d%d", &flag, &x);
if(flag)//来了领养者
{
if(s.empty())//没有宠物
s1.insert(x);
else
{
l = --s.lower_bound(x);
r = s.lower_bound(x);
if(r != s.begin() && r != s.end())
{
if(x - *l <= *r - x)cnt = l;
else cnt = r;
}
else if(r == s.begin())cnt = r;
else cnt = l;
ans += abs(*cnt - x);
ans %= ;
s.erase(cnt);
}
}
else//来了宠物
{
if(s1.empty())//没有领养者
s.insert(x);
else
{
l = --s1.lower_bound(x);
r = s1.lower_bound(x);
if(r != s1.begin() && r != s1.end())
{
if(x - *l <= *r - x)cnt = l;
else cnt = r;
}
else if(r == s1.begin())cnt = r;
else cnt = l;
ans += abs(*cnt - x);
ans %= ;
s1.erase(cnt);
}
}
}
cout<<ans<<endl;
return Accepted;
}