[LOJ2275][JXOI2017]颜色 线段树+优先队列

显然,任意一种合法的删颜色的方式都和一对合法的切环的断点一一对应。
然后就和BZOJ4382的第一问一样了。
//反正我就把一个 改成了 = <script type="math/tex" id="MathJax-Element-4">=</script>就过了
代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
using namespace std;
const int maxn=300010;
int n,c[maxn],pre[maxn],lst[maxn];
vector<int> a[maxn];
int read()
{
    int x=0;char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar());
    for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
    return x;
}
struct heap
{
    priority_queue<int> Q,D;
    void init()
    {
        while(!Q.empty()) Q.pop();
        while(!D.empty()) D.pop();
    }
    void push(int x) {Q.push(x);}
    void erase(int x) {D.push(x);}
    int top()
    {
        while(!D.empty()&&Q.top()==D.top()) Q.pop(),D.pop();
        return Q.empty()?0:Q.top();
    }
}H;
struct tree
{
    int l,r,sum;
    bool b,cvr;
    tree *ls,*rs;
    tree(){ls=rs=NULL;cvr=0;}
    void cal(){sum=0;b=0;}
    void update(){b=ls->b||rs->b; sum=ls->sum+rs->sum;}
    void pushdown()
    {
        if(cvr) ls->cvr=rs->cvr=1,ls->cal(),rs->cal(),cvr=0;
    }
    void build(int lx,int rx)
    {
        l=lx;r=rx;b=1;sum=r-l+1;
        if(l==r) return;
        int mid=(l+r)>>1;
        (ls=new tree)->build(lx,mid);
        (rs=new tree)->build(mid+1,rx);
    }
    void del(int lx,int rx)
    {
        if(lx>rx) return;
        if(lx==l&&rx==r) {cvr=1;cal();return;}
        pushdown();
        int mid=(l+r)>>1;
        if(rx<=mid) ls->del(lx,rx);
        else if(lx>mid) rs->del(lx,rx);
        else ls->del(lx,mid),rs->del(mid+1,rx);
        update();
    }
    int qry(int lx,int rx)
    {
        if(lx>rx) return 0;
        if(l==lx&&r==rx) return sum;
        pushdown();
        int mid=(l+r)>>1;
        if(rx<=mid) return ls->qry(lx,rx);
        else if(lx>mid) return rs->qry(lx,rx);
        else return ls->qry(lx,mid)+rs->qry(mid+1,rx);
    }
}*xtr;
int main()
{
    int ca=read();
    while(ca--)
    {
        n=read();
        for(int i=1;i<=n;i++)
            c[i]=read();
        (xtr=new tree)->build(1,n);
        for(int i=1;i<=n;i++)
            a[c[i]].clear();
        for(int i=n;i;i--)
            a[c[i]].push_back(i);
        H.init();   
        long long ans1=0;
        for(int i=1;i<=n;i++)
        {
            H.push(i);
            if(a[c[i]][0]==i)
            {
                for(int j=0;j<a[c[i]].size();j++)
                    H.erase(a[c[i]][j]);
                xtr->del(*(a[c[i]].end()-1)+1,a[c[i]][0]);   
            }
            int lft=H.top()+1;
            ans1+=xtr->qry(lft,i);
        }   
        printf("%lld\n",ans1);
    }   
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值