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);  
}