BZOJ1691: [Usaco2007 Dec]挑剔的美食家 Treap

BZOJ 1691: [Usaco2007 Dec]挑剔的美食家

Time Limit: 5 Sec  Memory Limit: 64 MB

Submit: 752  Solved: 370
[Submit][Status][Discuss]

题解：
把牛和草都按照鲜嫩程度降序排列，每次将大于一个牛要求的鲜嫩程度的草全部扔进平衡树中，平衡树按照价钱大小来建，然后对于每一头牛求出来他的后继并将其删除就可以了（如果找不到后继输出-1）。

Treap写法：

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=100005;
struct usaco{int v,d;}a
,b
;
bool cmp(usaco u,usaco v){return u.d>v.d;}
int n,m,p
,ls
,rs
,rnd
,w
,root;
long long ans;
void rx(int &x)
{
int t=ls[x];
ls[x]=rs[t],rs[t]=x;
x=t;
}
void lx(int &x)
{
int t=rs[x];
rs[x]=ls[t],ls[t]=x;
x=t;
}
int cnt;
void Insert(int &x,int v)
{
if(!x)
{x=++cnt,p[x]=v,w[x]=1,rnd[x]=rand();return;}
if(v==p[x]) w[x]++;
else if(v<p[x]) {Insert(ls[x],v);if(rnd[ls[x]]>rnd[x])rx(x);}
else {Insert(rs[x],v);if(rnd[rs[x]]>rnd[x])lx(x);}
}
void Del(int &x,int v)
{
if(p[x]==v)
{
if(w[x]>1) w[x]--;
else if(ls[x]*rs[x]==0) x=ls[x]+rs[x];
else if(rnd[ls[x]]<rnd[rs[x]]) lx(x),Del(x,v);
else rx(x),Del(x,v);
}
else if(v<p[x]) Del(ls[x],v);
else Del(rs[x],v);
}
int tmp;
void Hou(int x,int v)
{
if(!x) return;
if(p[x]>=v){tmp=p[x];Hou(ls[x],v);}
else Hou(rs[x],v);
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d%d",&a[i].v,&a[i].d);
for(int i=1;i<=m;i++) scanf("%d%d",&b[i].v,&b[i].d);
sort(a+1,a+n+1,cmp);
sort(b+1,b+m+1,cmp);
int j=1;
for(int i=1;i<=n;i++)
{
while(b[j].d>=a[i].d&&j<=m) Insert(root,b[j].v),j++;
tmp=-1;
Hou(root,a[i].v);
if(tmp==-1)
{
printf("-1");
return 0;
}
ans+=tmp;
Del(root,tmp);
}
printf("%lld",ans);
}

mutiset（可以有重复元素的set）。跑起来稍微快了一些，代码也非常的好写：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<set>
using namespace std;
const int N=100005;
struct dat{int v,d;}a
,b
;
bool cmp(dat u,dat v){return u.d>v.d;}
multiset<int>S;
int n,m;
long long ans;
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d%d",&a[i].v,&a[i].d);
for(int i=1;i<=m;i++) scanf("%d%d",&b[i].v,&b[i].d);
sort(a+1,a+n+1,cmp);
sort(b+1,b+m+1,cmp);
int j=1;
multiset<int>::iterator it;
for(int i=1;i<=n;i++)
{
while(a[i].d<=b[j].d&&j<=m) S.insert(b[j].v),j++;
it=S.lower_bound(a[i].v);
if(it==S.end()) {puts("-1");return 0;}
ans+=*it;
S.erase(it);
}
printf("%lld",ans);
}

Description

与很多奶牛一样，Farmer John那群养尊处优的奶牛们对食物越来越挑剔，随便拿堆草就能打发她们午饭的日子自然是一去不返了。现在，Farmer John不得不去牧草专供商那里购买大量美味多汁的牧草，来满足他那N(1 <= N <= 100,000)头挑剔的奶牛。 所有奶牛都对FJ提出了她对牧草的要求：第i头奶牛要求她的食物每份的价钱不低于A_i(1 <= A_i
<= 1,000,000,000)，并且鲜嫩程度不能低于B_i(1 <= B_i <= 1,000,000,000)。商店里供应M(1 <= M <= 100,000)种不同的牧草，第i 种牧草的定价为C_i(1 <= C_i <= 1,000,000,000)，鲜嫩程度为D_i (1 <= D_i <= 1,000,000,000)。 为了显示她们的与众不同，每头奶牛都要求她的食物是独一无二的，也就是说，没有哪两头奶牛会选择同一种食物。 Farmer John想知道，为了让所有奶牛满意，他最少得在购买食物上花多少钱。

Input

* 第1行: 2个用空格隔开的整数：N 和 M

* 第2..N+1行: 第i+1行包含2个用空格隔开的整数：A_i、B_i * 第N+2..N+M+1行: 第j+N+1行包含2个用空格隔开的整数：C_i、D_i

Output

* 第1行: 输出1个整数，表示使所有奶牛满意的最小花费。如果无论如何都无法 满足所有奶牛的需求，输出-1

Sample Input

4 7

1 1

2 3

1 4

4 2

3 2

2 1

4 3

5 2

5 4

2 6

4 4

Sample Output

12

输出说明:

给奶牛1吃价钱为2的2号牧草，奶牛2吃价钱为4的3号牧草，奶牛3分到价钱

为2的6号牧草，奶牛4选择价钱为4的7号牧草，这种分配方案的总花费是12，为

所有方案中花费最少的。

HINT

Source

Gold

[Submit][Status][Discuss]