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[N],b[N];
bool cmp(usaco u,usaco v){return u.d>v.d;}
int n,m,p[N],ls[N],rs[N],rnd[N],w[N],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[N],b[N];
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]