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

显然,任意一种合法的删颜色的方式都和一对合法的切环的断点一一对应。
然后就和BZOJ4382的第一问一样了。
//反正我就把一个 改成了 = 就过了
代码:

#include
#include
#include
#include
#include
#include
#include
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;jdel(*(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;
}

你可能感兴趣的:(数据结构,优先队列,线段树)