bzoj 5011: [Jx2017]颜色 单调栈

题意

可怜有一个长度为n的正整数序列Ai，其中相同的正整数代表着相同的颜色。现在可怜觉得这个序列太长了，于是她决定选择一些颜色把这些颜色的所有位置都删去。删除颜色i可以定义为把所有满足Aj=i的位置j都从序列中删去。然而有些时候删去之后，整个序列变成了好几段，可怜不喜欢这样，于是她想要知道有多少种删去颜色的方案使得最后剩下来的序列非空且连续。例如颜色序列{1,2,3,4,5}，删除颜色3后序列变成了{1,2}和{4,5}两段，不满足条件。而删除颜色1后序列变成了{2,3,4,5}，满足条件。两个方案不同当且仅当至少存在一个颜色i只在其中一个方案中被删去。
n<=300000

分析

一开始的想法是把每种颜色看做一个点，然后每种颜色向其最左边和最右边两个端点之间的颜色连边，用线段树优化建图然后tarjan缩点，这样问题就变成了求一个DAG的独立集数量。然后发现并不会求这个东西，然后就放弃了。
设每种颜色出现位置的最小值和最大值为mn[i]和mx[i]。
那么题目其实就是问有多少个区间[l,r]满足max(mx[l..r])<=r且min(mn[l,r])>=l。
这个可以枚举右端点，然后用一个单调栈来维护mx，用另外一个单调栈来维护所有满足mn限制的左端点，每次只要在单调栈上二分一下即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=300005;

int n,c[N],mn[N],mx[N],l[N],r[N],s1[N],s2[N];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int binary(int t1,int x)
{
    int L=1,R=t1;
    while (L<=R)
    {
        int mid=(L+R)/2;
        if (r[s1[mid]]>x) L=mid+1;
        else R=mid-1;
    }
    return s1[L-1]+1;
}

int main()
{
    int T=read();
    while (T--)
    {
        n=read();
        for (int i=1;i<=n;i++) mn[i]=n,mx[i]=1;
        for (int i=1;i<=n;i++) c[i]=read(),mn[c[i]]=min(mn[c[i]],i),mx[c[i]]=i;
        for (int i=1;i<=n;i++) l[i]=mn[c[i]],r[i]=mx[c[i]];
        LL ans=0;int t1=0,t2=0;
        for (int i=1;i<=n;i++)
        {
            while (t1&&r[s1[t1]]<=r[i]) t1--;
            s1[++t1]=i;s2[++t2]=i;
            while (t2&&s2[t2]>l[i]) t2--;
            int lim=binary(t1,i),r=lower_bound(s2+1,s2+t2+1,lim)-s2;
            ans+=t2-r+1;
        }
        printf("%lld\n",ans);
    }
    return 0;
}