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

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，为
所有方案中花费最少的。

 

题解

排序+平衡树。

由于有两个限制条件，我们可以通过排序满足其中一个条件。然而另一个条件是：大于某个数且最小。所以这里可以用平衡树。因为操作比较单一，所以用了treap。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#define MAXN 100002
#define ll long long
using namespace std;
int n,m;
struct niu {int c,v;} a[MAXN],b[MAXN];
struct shu {int l,r,v,rnd,cs;} tr[MAXN];
int root,size,tag;
ll ans;
bool kp(const niu &i,const niu &j) {return i.v>j.v;}
void init()
{
	scanf("%d%d",&n,&m);
	int i;
	for(i=1;i<=n;i++) scanf("%d%d",&a[i].c,&a[i].v);
	for(i=1;i<=m;i++) scanf("%d%d",&b[i].c,&b[i].v);
	sort(a+1,a+n+1,kp);
	sort(b+1,b+m+1,kp);
}
void lturn(int &w)
{
	int t=tr[w].r;
	tr[w].r=tr[t].l; tr[t].l=w; w=t;
}
void rturn(int &w)
{
	int t=tr[w].l;
	tr[w].l=tr[t].r; tr[t].r=w; w=t;
}
void insert(int &w,int x)
{
	if(!w)
	   {w=++size; tr[w].v=x; tr[w].cs=1; tr[w].rnd=rand(); return;}
	if(x==tr[w].v) tr[w].cs++;
	else if(x<tr[w].v)
	   {insert(tr[w].l,x);
	    if(tr[tr[w].l].rnd<tr[w].rnd) rturn(w);
	   }
	else
	   {insert(tr[w].r,x);
	    if(tr[tr[w].r].rnd<tr[w].rnd) lturn(w);
	   }
}
void find(int &w,int x)
{
	if(!w) return;
	if(tr[w].v>=x) {tag=tr[w].v; find(tr[w].l,x);}
	else find(tr[w].r,x);
}
void del(int &w,int x)
{
	int l=tr[w].l,r=tr[w].r;
	if(tr[w].v==x)
	   {if(tr[w].cs>1) tr[w].cs--;
	    else if(tr[w].l*tr[w].r==0) w=tr[w].l+tr[w].r;
	    else if(tr[l].rnd<tr[r].rnd) {rturn(w); del(w,x);}
	    else {lturn(w); del(w,x);}
	   }
	else if(x<tr[w].v) del(tr[w].l,x);
	else del(tr[w].r,x);
}
void work()
{
	int i,j=1;
	for(i=1;i<=n;i++)
	   {tag=-1;
		while(b[j].v>=a[i].v&&j<=m)
	       {insert(root,b[j].c); j++;}
	    find(root,a[i].c);
	    if(tag==-1) {printf("-1\n"); return;}
	    else
	       {ans+=tag; del(root,tag);} 
	   }
	printf("%lld\n",ans);
}
int main()
{
	init(); work();
	return 0;
}