传送门
解析:
话说
S
T
L
STL
STL不开
O
2
O2
O2真的慢啊。。。
开了
O
2
O2
O2跑得飞快。。。
思路:
首先是一个做多了题就懂的套路。
要求一条一条地删边,我们就直接倒序加边。。。
维护连通性是显然的并查集啊。
话说异或为0怎么处理?
这是一棵树啊。。。
异或是自反的啊。。。
那树上差分不就好了。。。
随便选一个根。
直接一趟
d
f
s
dfs
dfs处理出所有点到根节点的异或距离。
然后我们就知道,只有
d
i
s
t
[
u
]
=
=
d
i
s
t
[
v
]
dist[u]==dist[v]
dist[u]==dist[v]的点
u
,
v
u,v
u,v才能够组成点对。
那么我们就需要一种数据结构,来维护当前联通块有多少相同的权值。。。
平衡树啊,那不成这种题你选择手写权值线段树合并?
可以自己手写,也可以直接用
m
a
p
map
map。
当然自己手写的比较快就是了。
我们将该联通块的信息记录在并查集的根节点中。
每次合并两块采用启发式合并。
更新答案也很简单。就是增加了 ∑ m a [ u ] [ v a l ] ∗ m a [ v ] [ v a l ] \sum ma[u][val]*ma[v][val] ∑ma[u][val]∗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;
}