Wc2014 紫荆花之恋

传送门

终于把心头大恨切掉了……

不知道为啥以前的代码交到UOJ上会MLE……今天下午重写了一遍，然后就过了……(然而并不知道究竟发生了什么……)

这次是用bfs写的，果然bfs就是比dfs快啊……

http://uoj.ac/submission/138458

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int maxn=100010,maxk=50;
const double alpha=0.75;
struct node{
    static inline int randint(){
        static int a=12132,b=233333333,p=998244353,x=666666666;
        x=a*x+b;x%=p;
        return x<0?(x+=p):x;
    }
    int key,size,p;
    node *ch[2];
    node(int k):key(k),size(1),p(randint()){}
    inline void refresh(){size=ch[0]->size+ch[1]->size+1;}
}*null=new node(0),*root[maxn],*root1[maxn][maxk];
void addnode(int,int);
void destroy(int);
void rebuild(int,int,int);
int getcenter(int,int);
void getdis(int,int,int);
node *newnode(int);
void delnode(node*);
void insert(int,node*&);
int order(int,node*);
void destroy(node*);
void rot(node*&,int);
queue<node*>freenodes;
vector<int>G[maxn],W[maxn];
int p[maxn],depth[maxn],d[maxn][maxk],id[maxn][maxk];
int size[maxn],siz[maxn][maxk],q[maxn];
bool vis[maxn]={false};
long long ans=0;
int n,w[maxn],c;
int main(){
    null->size=0;
    null->ch[0]=null->ch[1]=null;
    scanf("%*d%d",&n);
    for(int i=0;i<=n;i++){
        root[i]=null;
        fill(root1[i],root1[i]+maxk,null);
    }
    scanf("%*d%*d%d",&w[1]);
    vis[1]=true;
    size[1]=1;
    insert(-w[1],root[1]);
    printf("0\n");
    for(int i=2;i<=n;i++){
        scanf("%d%d%d",&p[i],&c,&w[i]);
        p[i]^=(ans%1000000000);
        G[i].push_back(p[i]);
        W[i].push_back(c);
        G[p[i]].push_back(i);
        W[p[i]].push_back(c);
        addnode(i,c);
        printf("%lld\n",ans);
    }
    return 0;
}
void addnode(int x,int c){
    depth[x]=depth[p[x]]+1;
    size[x]=1;
    vis[x]=true;
    insert(-w[x],root[x]);
    int rt=0;
    for(int u=p[x],k=depth[x]-1;u;u=p[u],k--){
        d[x][k]=d[p[x]][k]+c;
        id[x][k]=id[p[x]][k]?id[p[x]][k]:x;
        ans+=order(w[x]-d[x][k]+1,root[u])-order(w[x]-d[x][k]+1,root1[id[x][k]][k]);
        insert(d[x][k]-w[x],root[u]);
        insert(d[x][k]-w[x],root1[id[x][k]][k]);
        size[u]++;
        siz[id[x][k]][k]++;
        if(siz[id[x][k]][k]>size[u]*alpha+5)rt=u;
    }
    if(rt){
        destroy(rt);
        rebuild(rt,size[rt],p[rt]);
    }
}
void destroy(int x){
    int head=0,tail=0;
    q[tail++]=x;
    vis[x]=false;
    while(head!=tail){
        x=q[head++];
        destroy(root[x]);
        root[x]=null;
        for(int i=depth[q[0]];i<=depth[x];i++){
            destroy(root1[x][i]);
            root1[x][i]=null;
            d[x][i]=id[x][i]=siz[x][i]=0;
        }
        for(int i=0;i<(int)G[x].size();i++)if(vis[G[x][i]]&&depth[G[x][i]]>=depth[q[0]]){
            vis[G[x][i]]=false;
            q[tail++]=G[x][i];
        }
    }
}
void rebuild(int x,int s,int pr){
    x=getcenter(x,s);
    vis[x]=true;
    p[x]=pr;
    depth[x]=depth[pr]+1;
    size[x]=s;
    insert(-w[x],root[x]);
    for(int i=0;i<(int)G[x].size();i++)if(!vis[G[x][i]]){
        p[G[x][i]]=x;
        d[G[x][i]][depth[x]]=W[x][i];
        getdis(G[x][i],x,depth[x]);
    }
    for(int i=0;i<(int)G[x].size();i++)if(!vis[G[x][i]])rebuild(G[x][i],size[G[x][i]],x);
}
int getcenter(int x,int s){
    int head=0,tail=0;
    q[tail++]=x;
    while(head!=tail){
        x=q[head++];
        size[x]=1;
        for(int i=0;i<(int)G[x].size();i++)if(!vis[G[x][i]]&&G[x][i]!=p[x]){
            p[G[x][i]]=x;
            q[tail++]=G[x][i];
        }
    }
    for(int i=s-1;i;i--)size[p[q[i]]]+=size[q[i]];
    x=q[0];
    for(;;){
        bool ok=false;
        for(int i=0;i<(int)G[x].size();i++)if(!vis[G[x][i]]&&G[x][i]!=p[x]&&(size[G[x][i]]<<1)>s){
            x=G[x][i];
            ok=true;
            break;
        }
        if(!ok)break;
    }
    return x;
}
void getdis(int x,int rt,int k){
    int head=0,tail=0;
    q[tail++]=x;
    while(head!=tail){
        x=q[head++];
        size[x]=1;
        id[x][k]=q[0];
        insert(d[x][k]-w[x],root[rt]);
        insert(d[x][k]-w[x],root1[q[0]][k]);
        for(int i=0;i<(int)G[x].size();i++)if(!vis[G[x][i]]&&G[x][i]!=p[x]){
            p[G[x][i]]=x;
            d[G[x][i]][k]=d[x][k]+W[x][i];
            q[tail++]=G[x][i];
        }
    }
    for(int i=tail-1;i;i--)size[p[q[i]]]+=size[q[i]];
    siz[q[0]][k]=size[q[0]];
}
node *newnode(int k){
    node *x;
    if(freenodes.empty())x=new node(k);
    else{
        x=freenodes.front();
        freenodes.pop();
        *x=node(k);
    }
    return x;
}
inline void delnode(node *x){freenodes.push(x);}
void insert(int x,node *&rt){
    if(rt==null){
        rt=newnode(x);
        rt->ch[0]=rt->ch[1]=null;
        return;
    }
    int d=x>rt->key;
    insert(x,rt->ch[d]);
    rt->refresh();
    if(rt->ch[d]->p<rt->p)rot(rt,d^1);
}
int order(int x,node *rt){
    int ans=0,d;
    while(rt!=null){
        if((d=x>rt->key))ans+=rt->ch[0]->size+1;
        rt=rt->ch[d];
    }
    return ans;
}
void destroy(node *x){
    if(x==null)return;
    destroy(x->ch[0]);
    destroy(x->ch[1]);
    delnode(x);
}
void rot(node *&x,int d){
    node *y=x->ch[d^1];
    x->ch[d^1]=y->ch[d];
    y->ch[d]=x;
    x->refresh();
    (x=y)->refresh();
}

话说在写之前还搞了一波事情，有兴趣的话可以看这里……