传送门
看了网上的题解,主流做法似乎是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;
}