JZOJ【NOIP2011DAY1】选择客栈(hotel)

题目：

Description

丽江河边有n家很有特色的客栈，客栈按照其位置顺序从1到n编号。每家客栈都按照某一种色调进行装饰（总共k种，用整数0 ~ k-1表示），且每家客栈都设有一家咖啡店，每家咖啡店均有各自的最低消费。 两位游客一起去丽江旅游，他们喜欢相同的色调，又想尝试两个不同的客栈，因此决定分别住在色调相同的两家客栈中。晚上，他们打算选择一家咖啡店喝咖啡，要求咖啡店位于两人住的两家客栈之间（包括他们住的客栈），且咖啡店的最低消费不超过p。 他们想知道总共有多少种选择住宿的方案，保证晚上可以找到一家最低消费不超过p元的咖啡店小聚。

Input

输入文件hotel.in，共n+1行。 第一行三个整数n，k，p，每两个整数之间用一个空格隔开，分别表示客栈的个数，色调的数目和能接受的最低消费的最高值； 接下来的n行，第i+1行两个整数，之间用一个空格隔开，分别表示i号客栈的装饰色调和i号客栈的咖啡店的最低消费。

Output

输出只有一行，一个整数，表示可选的住宿方案的总数。

Sample Input

5 2 3

0 5

1 3

0 2

1 4

1 5

Sample Output

3

Data Constraint

Hint

2人要住同样色调的客栈，所有可选的住宿方案包括：住客栈①③，②④，②⑤，④⑤，但是若选择住4、5号客栈的话，4、5号客栈之间的咖啡店的最低消费是4，而两人能承受的最低消费是3元，所以不满足要求。因此只有前3种方案可选。

对于30%的数据，有n≤100； 对于50%的数据，有n≤1,000； 对于100%的数据，有2≤n≤200,000，0

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int INF=0x7f7f7f;
struct node1
{
int color;
int cost;
}hotel[200005];
struct node2
{
int right;
int left;
int sum;
}tree[1000001];
int n,k,p;
int ans=0;
int s,c;
int num[60];
int cc[60][200001];
void build(int now,int l,int r)
{
tree[now].left=l;
tree[now].right=r;
if(l==r)
{
tree[now].sum=hotel[l].cost;
return;
}
int mid=(l+r)/2;
build(now*2,l,mid);
build(now*2+1,mid+1,r);
tree[now].sum=min(tree[now*2].sum,tree[now*2+1].sum);
}
int search(int now,int l,int r)
{
if(tree[now].left>r||tree[now].right<l)
return INF;
if(tree[now].left>=l&&tree[now].right<=r) return tree[now].sum;
return min(search(now*2,l,r),search(now*2+1,l,r));
}
int main()
{

scanf("%d%d%d",&n,&k,&p);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&c,&s);
hotel[i].color=c;
hotel[i].cost=s;
num[c]++;
cc[c][num[c]]=i;
}
build(1,1,n);
for(int i=0;i<k;i++)
{
for(int j=1;j<num[i];j++)
{
for(int q=j+1;q<=num[i];q++)
{
int sum=search(1,cc[i][j],cc[i][q]);
if(sum<=p)
{
ans+=num[i]-q+1;
break;
}
}
}
}
printf("%d",ans);
}