BZOJ1095(动态点分治+堆)

终于把这个坑填了..

按重心分治建树,每个点存两个堆,第一个存的是这个点子树中的点到父重心的距离,第二个存的是子节点第一个堆的堆顶,同时有一个全局答案堆,存的是每个点第二个堆的最大值+次大值。

20亿个蒟蒻们给了我一份漏洞百出的代码,害的我想了半天..

#include <cstdio>
#include <queue>
using namespace std;

const int N=100005,M=200005;
char op[5];
int n,e,x,y,q,tt,sm,mn,rt,hd[N],nx[M],to[M],v[N],f[N],a[N],d[N],sz[N],g[N][17];
struct hep {
    priority_queue<int> ad,dl;
    void pu(int x) {ad.push(x);}
    void po(int x) {dl.push(x);}
    int tp() {while(!dl.empty()&&ad.top()==dl.top()) ad.pop(),dl.pop(); return ad.top();}
    int gt() {int a=tp(); ad.pop(); int b=tp(); ad.push(a); return a+b;}
}a1,q1[N],q2[N];
void ad(int x,int y) {to[++e]=y,nx[e]=hd[x],hd[x]=e;}

void dfs2(int x) {for(int i=hd[x];i;i=nx[i]) if(!d[to[i]]) d[to[i]]=d[x]+1,g[to[i]][0]=x,dfs2(to[i]);}
int lca(int x,int y) {
    if(d[x]<d[y]) swap(x,y);
    for(int i=16;~i;i--) if(d[g[x][i]]>=d[y]) x=g[x][i];
    if(x==y) return x;
    for(int i=16;~i;i--) if(g[x][i]^g[y][i]) x=g[x][i],y=g[y][i];
    return g[x][0];
}
int ds(int x,int y) {return d[x]+d[y]-d[lca(x,y)]*2;}
void dfs(int x,int fa) {
    sz[x]=1;
    for(int i=hd[x];i;i=nx[i]) if(to[i]!=fa&&!v[to[i]]) dfs(to[i],x),sz[x]+=sz[to[i]];
}
void gt(int x,int fa) {
    int mx=0; sz[x]=1;
    for(int i=hd[x];i;i=nx[i]) if(to[i]!=fa&&!v[to[i]]) gt(to[i],x),sz[x]+=sz[to[i]],mx=max(mx,sz[to[i]]);
    mx=max(mx,tt-sz[x]);
    if(mx<mn) mn=mx,rt=x;
}
void bd(int x,int y) {
    gt(x,0),f[rt]=y,v[rt]=1,dfs(rt,0); int k=rt;
    for(int i=hd[rt];i;i=nx[i]) if(!v[to[i]]) tt=mn=sz[to[i]],bd(to[i],k);
}
void cal(int x) {
    int u=x;
    if(q2[x].ad.size()-q2[x].dl.size()>=2) a1.po(q2[x].gt());
    q2[x].pu(0);
    if(q2[x].ad.size()-q2[x].dl.size()>=2) a1.pu(q2[x].gt());
    while(f[u]) {
        if(q2[f[u]].ad.size()-q2[f[u]].dl.size()>=2) a1.po(q2[f[u]].gt());
        if(q1[u].ad.size()-q1[u].dl.size()) q2[f[u]].po(q1[u].tp());
        q1[u].pu(ds(f[u],x)),q2[f[u]].pu(q1[u].tp());
        if(q2[f[u]].ad.size()-q2[f[u]].dl.size()>=2) a1.pu(q2[f[u]].gt());
        u=f[u];
    }
}
void cal2(int x) {
    int u=x;
    if(q2[x].ad.size()-q2[x].dl.size()>=2) a1.po(q2[x].gt());
    q2[x].po(0);
    if(q2[x].ad.size()-q2[x].dl.size()>=2) a1.pu(q2[x].gt());
    while(f[u]) {
        if(q2[f[u]].ad.size()-q2[f[u]].dl.size()>=2) a1.po(q2[f[u]].gt());
        q2[f[u]].po(q1[u].tp()),q1[u].po(ds(f[u],x));
        if(q1[u].ad.size()-q1[u].dl.size()) q2[f[u]].pu(q1[u].tp());
        if(q2[f[u]].ad.size()-q2[f[u]].dl.size()>=2) a1.pu(q2[f[u]].gt());
        u=f[u];
    }
}

int main() {
    scanf("%d",&n),sm=n;
    for(int i=1;i<n;i++) scanf("%d%d",&x,&y),ad(x,y),ad(y,x);
    tt=mn=n,bd(1,0),d[1]=1,dfs2(1);
    for(int j=1;j<17;j++)
    for(int i=1;i<=n;i++)
        g[i][j]=g[g[i][j-1]][j-1];
    for(int i=1;i<=n;i++) cal(i);
    scanf("%d",&q);
    while(q--) {
        scanf("%s",op);
        if(op[0]=='G') {
            if(sm<=1) printf("%d\n",sm-1);
            else printf("%d\n",a1.tp());
        }
        else {
            scanf("%d",&x),a[x]^=1;
            if(a[x]) sm--,cal2(x); else sm++,cal(x);
        }
    }
    return 0;
}
posted @ 2017-03-04 20:59  Monster_Yi  阅读(302)  评论(2编辑  收藏  举报