2018.09.26【洛谷P3359】改造异或树（并查集）（启发式合并）

传送门

---

解析：

话说$STL$不开$O2$真的慢啊。。。
开了$O2$跑得飞快。。。

思路：

首先是一个做多了题就懂的套路。
要求一条一条地删边，我们就直接倒序加边。。。
维护连通性是显然的并查集啊。

话说异或为0怎么处理？
这是一棵树啊。。。
异或是自反的啊。。。
那树上差分不就好了。。。

随便选一个根。
直接一趟$dfs$处理出所有点到根节点的异或距离。
然后我们就知道，只有$dist[u]==dist[v]$的点$u,v$才能够组成点对。
那么我们就需要一种数据结构，来维护当前联通块有多少相同的权值。。。

平衡树啊，那不成这种题你选择手写权值线段树合并？
可以自己手写，也可以直接用$map$。

当然自己手写的比较快就是了。

我们将该联通块的信息记录在并查集的根节点中。
每次合并两块采用启发式合并。

更新答案也很简单。就是增加了$\sum ma[u][val]\ast ma[v][val]$，这个很好推导。

这道题就做完了。

---

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define gc getchar
#define pc putchar
#define cs const

inline
ll getint(){
    re ll num;
    re char c;
    re bool f=0;
    while(!isdigit(c=gc()))f^=c=='-';num=c^48;
    while(isdigit(c=gc()))num=(num<<1)+(num<<3)+(c^48);
    return f?-num:num;
}

inline
void outint(ll a){
    static char ch[23];
    if(a==0)pc('0');
    if(a<0)pc('-'),a=-a;
    while(a)ch[++ch[0]]=a-a/10*10,a/=10;
    while(ch[0])pc(ch[ch[0]--]^48);
}

cs int N=100005;
int n;
map<int,int> ma[N];
int fa[N];
ll ans;

inline
int getfa(int x){return x==fa[x]?x:fa[x]=getfa(fa[x]);}

inline
void merge(int u,int v){
    u=getfa(u);
    v=getfa(v);
    if(ma[u].size()>ma[v].size())swap(u,v);
    for(map<int,int>::iterator it=ma[u].begin();it!=ma[u].end();++it){
        int val=it->first;
        map<int,int>::iterator pos=ma[v].find(val);
        int cnt=it->second;
        if(pos!=ma[v].end())cnt+=pos->second,ans+=1ll*it->second*pos->second;
        ma[v][val]=cnt;
    }
    ma[u].clear();
    fa[u]=v;
}

int last[N],nxt[N<<1],to[N<<1],w[N<<1],ecnt;
struct edge{int u,v;}E[N];
inline
void addedge(int u,int v,int val){
    nxt[++ecnt]=last[u],last[u]=ecnt,to[ecnt]=v,w[ecnt]=val;
    nxt[++ecnt]=last[v],last[v]=ecnt,to[ecnt]=u,w[ecnt]=val;
    E[ecnt>>1]=edge{u,v};
}

int dist[N];
inline
void dfs(int u,int fa){
    for(int re e=last[u],v=to[e];e;v=to[e=nxt[e]]){
        if(v==fa)continue;
        dist[v]=dist[u]^w[e];
        dfs(v,u);
    }
}

int order[N];
ll Ans[N];

signed main(){
    n=getint();
    for(int re i=1;i<n;++i){
        int u=getint(),v=getint(),val=getint();
        addedge(u,v,val);
    }
    for(int re i=1;i<n;++i)order[i]=getint();
    dfs(1,0);
    for(int re i=1;i<=n;++i)ma[i][dist[i]]=1,fa[i]=i;
    for(int re i=n;i>=1;--i){
        int e=order[i];
        int u=getfa(E[e].u),v=getfa(E[e].v);
        merge(u,v);
        Ans[i]=ans;
    }
    for(int re i=1;i<=n;++i){
        outint(Ans[i]),pc('\n');
    }
    return 0;
}