传送门
看了网上的题解,主流做法似乎是dp,然而我做题时完全没有往那上面想。。
我是用 相同色调客栈两两配对的方案数-不符合条件的方案数
以样例为例。红色为颜色0,有2个;蓝色为颜色1,有3个
则相同色调客栈两两配对的方案数=2*(2-1)/2+3*(3-1)/2=4
w[i]为咖啡店最低消费,1、4、5为无法使用的咖啡店
首先看1,这块区域没有可以两两配对的咖啡店
区域4~5,有一对相同色调的咖啡店4、5
所以答案为4-1=3
接下去是超级慢的代码orz
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=200001;
int n,k,p,col[N],w[N+1],L[N+1],t;
long long ans;
int main(){
scanf("%d%d%d",&n,&k,&p);
for(int i=1;i<=n;i++) scanf("%d%d",&col[i],&w[i]);
for(int i=0;i<k;i++){
memset(L,0,sizeof L);
for(int j=1;j<=n;j++) L[j]=L[j-1]+(col[j]==i);//用前缀和维护一段区间内相同色调的咖啡店个数
ans+=(long long) L[n]*(L[n]-1)/2;//加上相同色调客栈两两配对的方案数
t=0;
for(int j=1;j<=n;j++)
if (w[j]>p&&w[j+1]<=p){
ans-=(long long) (L[j]-L[t])*(L[j]-L[t]-1)/2;//减去不符合条件的方案数
} else
if (w[j+1]>p&&w[j]<=p) t=j;//t为一段'>'区间内起始位置
}
printf("%lld",ans);
return 0;
}