BZOJ 3319: 黑白树 树+并查集+未调完+神题

Code:

#include<bits/stdc++.h>
#define maxn 1000003 
using namespace std;
char *p1,*p2,buf[100000];
#define nc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++)
int rd() {int x=0,f=1; char c=nc(); while(c<48) {if(c=='-') f=-1; c=nc();} while(c>47) x=(((x<<2)+x)<<1)+(c^48),c=nc(); return x*f;}
void setIO(string s)
{
    string in=s+".in", out=s+".out"; 
    freopen(in.c_str(),"r",stdin); 
    freopen(out.c_str(),"w",stdout); 
} 
int n,Q,edges; 
int hd[maxn],to[maxn<<1],nex[maxn<<1],val[maxn<<1],fa[maxn],dep[maxn],p[maxn],mark[maxn],tag[maxn]; 
int pd[maxn], answer[maxn]; 
void Init()
{
    for(int i=0;i<maxn;++i) p[i]=i; 
}         
int find(int x)
{
    return p[x]==x?x:p[x]=find(p[x]);   
}
void addedge(int u,int v,int c)
{
    nex[++edges]=hd[u],hd[u]=edges,to[edges]=v,val[edges]=c;  
}
void dfs(int u,int ff)
{
    fa[u]=ff; 
    for(int i=hd[u];i;i=nex[i])
    {
        int v=to[i]; 
        if(v==ff) continue;  
        dep[v]=dep[u]+1; 
        mark[v]=val[i];             
        dfs(v,u); 
    }
}
struct OPT
{
    int o,u,v; 
}opt[maxn];   
void _2(int u,int v,int cur)
{
    u=find(u),v=find(v); 
    while(u!=v)
    {
        // v is deeper than u 
        if(dep[u] > dep[v]) swap(u,v);  
        if(!pd[v]) p[v]=find(fa[v]), pd[v]=cur; 
       //  printf("%d %d\n",dep[u],dep[v]); 
        v = p[v]; 
    }
}
//set u -> v to white 
void solve(int u,int v,int cur)
{
    u=find(u),v=find(v); 
    while(u!=v)     
    {
        if(dep[u] > dep[v]) swap(u,v); 
        if(pd[v]==cur) p[v]=find(fa[v]);                     
        v = p[fa[v]];                  // 暴力跳QAQ......    
        //v=p[v]; 
    }
}             
int main()
{
  //   setIO("input"); 
   // scanf("%d%d",&n,&Q);
    n=rd(),Q=rd(); 
    for(int i=1;i<n;++i)
    {
        int u,v; 
        u=rd(),v=rd(); 
        //scanf("%d%d",&u,&v); 
        addedge(u,v,i);
        addedge(v,u,i); 
    }
    dep[1]=1; 
    dfs(1,0);          
    for(int i=1;i<=Q;++i)
    {
        opt[i].o=rd(); 
        //scanf("%d",&opt[i].o); 
        if(opt[i].o==1)  opt[i].u=rd(); 
           // scanf("%d",&opt[i].u); 
        else opt[i].u=rd(), opt[i].v=rd(); 
           // scanf("%d%d",&opt[i].u,&opt[i].v); 
    }
    // 处理黑点情况. 
    Init();             
    for(int i=1;i<=Q;++i)
    { 
        if(opt[i].o==2) _2(opt[i].u,opt[i].v,i); 
    }
    Init(); 
    for(int i=2;i<=n;++i) 
    {
        if(!pd[i])               //到最后也未被染成黑色 直接用并查集连上   
        {    
            int u = fa[i], v = i; 
            int x = find(u); 
            p[v] = x; 
        }
    }         
    int tot = 0; 
    for(int i=Q;i>=1;--i)
    {
        if(opt[i].o==2)
        {
            //debug(); 
            solve(opt[i].u, opt[i].v, i);    
            // debug(); 
        }
        else 
        {
            int x = find(opt[i].u);   
            answer[++tot]=mark[x];           
        }
    }
    for(int i=tot;i>=1;--i) printf("%d\n",answer[i]); 
    return 0; 
}