【Noip2011】选择客栈

【Noip2011】选择客栈

---

描述：

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

---

输入：

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

---

输出：

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

---

题解：

暴力枚举肯定超时，所以我们要用点优化。双端枚举+线段树会卡在1000ms左右，可以用vector之内的把颜色相同的客栈位置存起来，循环枚举左边客栈找右边客栈。注意咖啡消费低于p的数量要用前缀和数组优化一下，不然容易超时（窝就是因为这个超时了十几次QAQ）。不放心的话还可以加点读入优化和输出优化。

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
int n,k,p;
int col[200010];
int val[200010];
vector<int> x[50];
int pla[50];
inline int read(){
    int X=0,w=1;
    char ch=0;
    while(ch<'0'||ch>'9'){
        if(ch=='-') w=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9'){
        X=(X<<3)+(X<<1)+ch-'0';
        ch=getchar();
    }
    return X*w;
}
int buf[30];
void write(int x)
{
    if (x<0) putchar('-'),x=-x;
    buf[0]=0;
    while (x) buf[++buf[0]]=x%10,x/=10;
    if (!buf[0]) buf[0]=1,buf[1]=0;
    while (buf[0]) putchar('0'+buf[buf[0]--]);
}
int pl[200010];
int main(){
    n=read();
    k=read();
    p=read();   
    for(register int i=1;i<=n;++i){
        col[i]=read();
        val[i]=read();
        x[col[i]].push_back(i);
        pl[i]=pl[i-1];
        if(val[i]<=p) pl[i]++;
    }
    int ans=0;
    int len;
    for(register int i=1;i<n;++i){
        len=x[col[i]].size();
        pla[col[i]]++;
        for(register int j=pla[col[i]];j<len;++j){
            if(pl[x[col[i]][j]]-pl[i-1]>0) ans++;
        }
    }
    write(ans);
    return 0;
}