标签:题目 答案 价格 P20 noip 咖啡 efi 选择 ast
题目描述
丽江河边有n家很有特色的客栈,客栈按照其位置顺序从1到n编号。每家客栈都按照某一种色调进行装饰(总共k种,用整数0~ k-1表示),且每家客栈都设有一家咖啡店,每家咖啡店均有各自的最低消费。
两位游客一起去丽江旅游,他们喜欢相同的色调,又想尝试两个不同的客栈,因此决定分别住在色调相同的两家客栈中。晚上,他们打算选择一家咖啡店喝咖啡,要求咖啡店位于两人住的两家客栈之间(包括他们住的客栈),且咖啡店的最低消费不超过p。
他们想知道总共有多少种选择住宿的方案,保证晚上可以找到一家最低消费不超过p元的咖啡店小聚。
输入
第一行三个整数n,k,p,每两个整数之间用一个空格隔开,分别表示客栈的个数,色调的数目和能接受的最低消费的最高值;
接下来的n行,第i+1行两个整数,之间用一个空格隔开,分别表示i号客栈的装饰色调和i号客栈的咖啡店的最低消费。
输出
输出只有一行,一个整数,表示可选的住宿方案的总数。
样例输入
5 2 3
0 5
1 3
0 2
1 4
1 5
样例输出
3
题解
这里是一种O(n)的极其优秀的方法。枚举第二个客栈,cnt[ i ]表示 i 颜色已经出现的次数,last[ i ]表示颜色 i 上一次出现的位置,now表示从当前位置向前寻找到颜色相同的价格小于p的第一个位置,sum[ i ]是一个临时存答案的数组。
#include<cmath> #include<cstdio> #include<cstdlib> #include<cstring> #include<iostream> #include<algorithm> using namespace std; #define ll long long const int maxn=200000+50; int n,k,p,last[maxn],cnt[maxn],sum[maxn],ans,now,col,pri; int main(){ cin>>n>>k>>p; for(int i=1;i<=n;i++){ cin>>col>>pri; if(pri<=p) now=i; if(now>=last[col]) sum[col]=cnt[col]; last[col]=i; ans+=sum[col]; cnt[col]++; } cout<<ans<<endl; return 0; }
标签:题目 答案 价格 P20 noip 咖啡 efi 选择 ast
原文地址:https://www.cnblogs.com/rlddd/p/9594946.html