bzoj1691[Usaco2007 Dec]挑剔的美食家 平衡树treap
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,为
所有方案中花费最少的。
题解:
题目要求的是 每个奶牛都找到一个唯一的材料,并且总花费最小我们检索按照花费排序之后对于i号奶牛 我们把 满足鲜嫩程度的 材料的定价 加入treap中
对于i号号奶牛的最优答案就是 a[i].a的前驱,也就是第一个大于等于a[i].a的数 用treap的前驱查找函数查找即可
加入找不到就是没有满足i号奶牛的材料跳出-1
这样每个材料都只遍历一次,奶牛遍历一次,查询logn
复杂度就是max(n,m)*log(n)
#include<iostream> #include<cstdio> #include<cstdlib> #include<cmath> #include<algorithm> #include<map> using namespace std; const int N = 1e5+10, mod =1e9+7; struct data{ int l,r,v,rnd,w; }tr[520005]; struct ss{int a,b;}a[N],b[N]; bool cmp (ss s1,ss s2) {return s1.b>s2.b;} int n,size,root,ans; void rturn(int &k) { int t=tr[k].l;tr[k].l=tr[t].r;tr[t].r=k;k=t; } void lturn(int &k) { int t=tr[k].r;tr[k].r=tr[t].l;tr[t].l=k;k=t; } void insert(int &k,int x) { if(k==0){ k=++size; tr[k].w=1; tr[k].v=x; tr[k].rnd=rand(); return; } if(tr[k].v==x)tr[k].w++; else if(x>tr[k].v) { insert(tr[k].r,x); if(tr[tr[k].r].rnd<tr[k].rnd)lturn(k); } else { insert(tr[k].l,x); if(tr[tr[k].l].rnd<tr[k].rnd)rturn(k); } } void del(int &k,int x) { if(k==0)return; if(tr[k].v==x) { if(tr[k].w>1){tr[k].w--;return;} else if(tr[k].l*tr[k].r==0)k=tr[k].l+tr[k].r; else if(tr[tr[k].l].rnd<tr[tr[k].r].rnd)rturn(k),del(k,x); else lturn(k),del(k,x); } else if(x>tr[k].v) del(tr[k].r,x); else del(tr[k].l,x); } void query_sub(int k,int x) { if(k==0)return; if(tr[k].v>=x) { ans=tr[k].v;query_sub(tr[k].l,x); } else query_sub(tr[k].r,x); } int m; int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) scanf("%d%d",&a[i].a,&a[i].b); for(int i=1;i<=m;i++) scanf("%d%d",&b[i].a,&b[i].b); sort(a+1,a+n+1,cmp); sort(b+1,b+m+1,cmp); int cnt = 1; long long L = 0; for(int i=1;i<=n;i++) { ans = -1; while(b[cnt].b>=a[i].b&&cnt<=m) {insert(root,b[cnt++].a);} query_sub(root,a[i].a); if(ans == -1) {puts("-1");return 0;} else { L += ans; del(root,ans); } } printf("%lld\n",L); }