bzoj 5011: [Jx2017]颜色

最新推荐文章于 2019-02-24 19:57:39 发布

原创最新推荐文章于 2019-02-24 19:57:39 发布 · 307 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

思路同时被 2 个专栏收录

37 篇文章

订阅专栏

单调栈

2 篇文章

订阅专栏

探讨了一个有趣的问题：如何从一个由颜色编号组成的序列中删除某些颜色，使得剩余的颜色形成一个连续且非空的序列。文章详细介绍了算法思路，包括使用单调栈来确定颜色的有效范围，并通过二分查找来优化解决方案。

题意：

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

题解：

因为太菜，连模型都没能转化……没救了。
其实是等价选一些数，他们是连续的一段区间。
即设 $L,R$ 为每种颜色出现的最左边与最右边。
问有多少区间[l,r]满足 $R[a[i]]<=r$ 且 $L[a[i]]>=l(l<=i<=r)$
维护两个单调栈，一个确定R的限制，另一个维护合法的左端点，然后~~在单调栈二分~~直接维护就好了。
code：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL long long
using namespace std;
int T,n,a[300010];
int L[300010],R[300010];
int s1[300010],s2[300010],t1,t2;//s1 关于L的的单调栈，s2 关于R的单调栈
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 solve(int x)
{
    int l=1,r=t1,ans=t1+1;
    while(l<=r)
    {
        int mid=(l+r)/2;
        if(s1[mid]>=x) ans=mid,r=mid-1;
        else l=mid+1;
    }
    return ans;
}
int main()
{
    T=read();
    while(T--)
    {
        LL ans=0;
        n=read();
        for(int i=1;i<=n;i++) a[i]=read();
        memset(L,0,sizeof(L));memset(R,0,sizeof(R));
        for(int i=1;i<=n;i++) if(!L[a[i]]) L[a[i]]=i;
        for(int i=n;i>=1;i--) if(!R[a[i]]) R[a[i]]=i;
        t1=t2=0;
        for(int i=1;i<=n;i++)
        {
            while(t2!=0&&R[a[s2[t2]]]<=R[a[i]]) t2--;s2[++t2]=i;
            while(t1!=0&&s1[t1]>L[a[i]]) t1--;
            if(L[a[i]]==i) s1[++t1]=i;
            if(R[a[i]]==i)
            {
                while(t2!=0&&R[a[s2[t2]]]<=i) t2--;
                int l=s2[t2]+1,k=solve(l);
                ans+=(LL)(t1-k+1);
            }
        }
        printf("%lld\n",ans);
    }
}