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
1 1
2 3
1 4
4 2
3 2
2 1
4 3
5 2
5 4
2 6
4 4
Sample Output
输出说明:
给奶牛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;
}