a[color] 记录最后一个颜色为 color 的客栈位置
num[color] 记录当前颜色为 color 的客栈数量
last为距离 i 最近一个符合消费的客栈位置
g[color]为 受last限制下能够符合的方案数
仅当 a[color]<=last 时 g[color]可更新为a[color]前所有的颜色为color的客栈数即num[color]
否则g[color]就只为上一次a[color]<=last时颜色为color的数量
关于g数组我写的有点迷幻,可以结合样例自己感觉一下。。。
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> int i,n,k,p,color,cost,last,ans,a[55],num[55],g[55]; int main() { scanf("%d%d%d",&n,&k,&p); for (i=1;i<=n;i++) { scanf("%d%d",&color,&cost); if (cost<=p) last=i; if (last>=a[color]) g[color]=num[color]; ans+=g[color]; num[color]++; a[color]=i; } printf("%d",ans); }