题目描述
丽江河边有\(n\)家很有特色的客栈,客栈按照其位置顺序从\(1\)到\(n\)编号。每家客栈都按照某一种色调进行装饰(总共\(k\)种,用整数\(0 \sim k−1\)表示),且每家客栈都设有一家咖啡店,每家咖啡店均有各自的最低消费。
两位游客一起去丽江旅游,他们喜欢相同的色调,又想尝试两个不同的客栈,因此决定分别住在色调相同的两家客栈中。晚上,他们打算选择一家咖啡店喝咖啡,要求咖啡店位于两人住的两家客栈之间(包括他们住的客栈),且咖啡店的最低消费不超过\(p\)。
他们想知道总共有多少种选择住宿的方案,保证晚上可以找到一家最低消费不超过\(p\)元的咖啡店小聚。
输入输出格式
输入格式:
共\(n+1\)行。
第一行三个整数\(n,k,p\),每两个整数之间用一个空格隔开,分别表示客栈的个数,色调的数目和能接受的最低消费的最高值;
接下来的\(n\)行,第\(i+1\)行两个整数,之间用一个空格隔开,分别表示\(i\)号客栈的装饰色调和\(i\)号客栈的咖啡店的最低消费。
输出格式:
一个整数,表示可选的住宿方案的总数。
输入输出样例
输入样例#1:
5 2 3
0 5
1 3
0 2
1 4
1 5
输出样例#1:
3
分析
判断一对客栈的可行性只与它们之间的是否存在最低消费不大于\(p\)的客栈,因此我们用\(last\)变量记录上一个满足条件的客栈,\(pos[i]\)记录上一个色调为\(i\)的客栈,\(cnt[i]\)记录色调\(i\)的客栈数量。只要\(last\)在\(pos[i]\)的后面,则当前客栈与之前的所有客栈的组合都可行,并以\(bfe[]\)数组记下此时的\(cnt[]\)。否则,当前客栈与上次\(bfe[]\)记录的客栈中间的客栈是不可行的,这时\(bfe\)数组便派上用场了,直接加上\(bfe[i]\)即可。
代码
#include<cstdio>
#include<cstring>
#include<iostream>
#define il inline
#define re register
#define maxn 200011
#define mymax(a,b) a>b?a:b
#define mymin(a,b) a<b?a:b
#define tie0 cin.tie(0),cout.tie(0)
#define fastio ios::sync_with_stdio(false)
using namespace std;
typedef long long ll;
template<typename T>inline void read(T &x){
T f=1;x=0;char c;
for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-1;
for(;isdigit(c);c=getchar())x=x*10+(c^48);
x*=f;
}
int n,k,p;
int last,sum;
int cnt[51],pos[51],bfe[51];
int main(){
fastio,tie0;
int a,b;
cin>>n>>k>>p;
for(int i=1;i<=n;++i){
cin>>a>>b;
if(b<=p) last=i;
if(last>=pos[a]) bfe[a]=cnt[a];
sum+=bfe[a];
pos[a]=i;
cnt[a]++;
}
printf("%d",sum);
return 0;
}