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