[BZOJ4382][POI2015]Podział naszyjnika 优先队列+线段树 hash+Twopoints

听说hash做法很简单。。。不管了,我写的是数据结构。。。
以下不要计较开闭区间等等。
考虑从左往右枚举右切口 r ,假设左切口 l 合法,那么对于每种颜色,要么分布在 [1,l][r,n] ,要么分布在 [l,r]
对于满足第一种情况的颜色,记 prec 为其 r 在之前最后一次出现的位置,那么要满足 l>max{prec} ,可以用优先队列维护(至于为什么先往后看)。
对于满足第二种情况的颜色,记 firstc lastc 为其第一次和最后一次出现的位置,那么首先要满足 r>=lastc ,然后要在优先队列中删除 c 出现的那些位置,然后 l 是不能在 [firstc,lastc] 之中的,可以把这个区间标记一下。
于是问题就是维护一个带删除操作的优先队列,并且会不断给一些区间标记,并且询问一个区间 [max{prec},r] 中有多少个未标记的点,用线段树即可。
对于第二问,我们考虑尽量让切的长度接近 n/2 ,于是就在 rn/2 左右找到第一个未被标记得点即可,也是线段树的经典操作。
代码:

#include
#include
#include
#include
#include
#include
#include
using namespace std;
const int maxn=1000010;
int n,c[maxn],pre[maxn],lst[maxn],k;
vector<int> a[maxn];
struct heap
{
    priority_queue<int> Q,D;
    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 find(int pl,bool f)
    {
        if(!b) return -1;
        if(l==r) return l;

        pushdown();
        int mid=(l+r)>>1,tmp=-1;
        if(f)
        {
            if(pl>mid) tmp=rs->find(pl,f);
            if(tmp==-1) tmp=ls->find(pl,f);
        }
        else
        {
            if(pl<=mid) tmp=ls->find(pl,f);
            if(tmp==-1) tmp=rs->find(pl,f);
        }
        return tmp;
    }
    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()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        scanf("%d",&c[i]);
    (xtr=new tree)->build(1,n);
    for(int i=n;i;i--)
        a[c[i]].push_back(i);
    long long ans1=0;int ans2=n;
    for(int i=1;iif(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,fl=xtr->find(max(lft,i-n/2+1),1),fr=xtr->find(max(lft,i-n/2+1),0);
        ans1+=xtr->qry(lft,i);
        if(fl>=lft) ans2=min(ans2,abs(n-2*(i-fl+1)));
        if(fr>=lft) ans2=min(ans2,abs(n-2*(i-fr+1)));
    }
    printf("%lld %d",ans1,ans2);
    return 0;
}

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