[VijosP1737][NOIp2011tg D1T2]选择客栈

客栈配对算法题解

最新推荐文章于 2024-06-15 16:15:35 发布

原创最新推荐文章于 2024-06-15 16:15:35 发布 · 244 阅读

0 ·

CC 4.0 BY-SA版权

不知道怎么分类专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种解决客栈配对问题的算法思路，通过计算相同色调客栈的两两配对方案数，并减去不符合条件的方案数来得出最终答案。文章提供了完整的C++代码实现。

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