NOIP 2011 day1 选择客栈

题目大意：

      

丽江河边有 n 家很有特色的客栈，客栈按照其位置顺序从1 到n 编号。每家客栈都按照

某一种色调进行装饰（总共k 种，用整数0 ~ k-1 表示），且每家客栈都设有一家咖啡店，每

家咖啡店均有各自的最低消费。

两位游客一起去丽江旅游，他们喜欢相同的色调，又想尝试两个不同的客栈，因此决定

分别住在色调相同的两家客栈中。晚上，他们打算选择一家咖啡店喝咖啡，要求咖啡店位于

两人住的两家客栈之间（包括他们住的客栈），且咖啡店的最低消费不超过p。

他们想知道总共有多少种选择住宿的方案，保证晚上可以找到一家最低消费不超过p

元的咖啡店小聚。

解法：

   

如果一家编号为 i 咖啡店的最低消费大于p，则记flag[ i ]=1;否则，flag[ i ]=0.

考虑到计算满足要求的对数不太容易， 从反面思考：

从1到n扫描出flag值等于1的连续客栈（在此区间内找到的客栈均为不满足要求的方案），在扫描时用cnt[j]记录当前颜色为j的客栈出现次数

cnt[j]++;

count+=(cnt [j]-1); （新的客栈可以与当前任意颜色相同客栈组合）

可以数出颜色相同但不满足要求的客栈对数。

答案ans=sum-count.(sum为所有颜色相同的客栈对数)

即为满足要求的客栈对数

代码：

#include<cstdio>#include<iostream>#include<algorithm>using namespace std;bool flag[200010];int cnt[100],Count,sum; int main(){int n,k,p,col[200010],tmp;scanf("%d%d%d",&n,&k,&p);for(int i=1;i<=n;i++){scanf("%d",&col[i]);scanf("%d",&tmp);if(tmp>p) flag[i]=1;}for(int i=1;i<=n;i++){if(flag[i]==0){memset(cnt,0,sizeof(cnt));continue;}cnt[col[i]]++;Count+=(cnt[col[i]]-1);}memset(cnt,0,sizeof(cnt));for(int i=1;i<=n;i++){cnt[col[i]]++;sum+=(cnt[col[i]]-1);}cout<<sum-Count<<endl;return 0;}