好吧,其实说是启发式合并,但是本题的真正难点还是在于map的定义上
map定义 :定义一个map< int , int > P[ x ] 表示从 以 x 为当前子树根节点向下遍历时,得到 异或和为 y 的情况数(有点绕),可以借助代码理解。因为我们知道 点a的异或值为 a^x1^x2^x3....^u 设为pre【a】,b的是 b^y1^y2^y3..^u
设为p【b】而答案要求p【a】^p【b】^u ==0(因为u异或了两次,需要再异或一次除去其中一次(异或法则))
虽说在这之前同样异或了u的父亲,父亲的父亲....但是异或后相同的值抵消并没有影响,所以这样是对的。
然后每次先运算再合并确保不会出现重复
启发式合并 : 每一次回溯的时候需要合并更新当前的map值,但是如果直接合并会T,所以每次合并的时候只把小的合并到大的上
提醒 我的编译器不支持for(auto a:P[u])只好用代送器慢慢遍历了,郁闷
#include<cstdio>
#include<cstring>
#include<iostream>
#include<map>
#define maxn 100020*2
#define ll long long
using namespace std;
map<int,int>P[maxn];
int head[maxn],cnt=1,n;
struct edge{
int u,next;
}e[maxn];
ll ans;
void adde(int a,int b){
e[cnt].u=b;
e[cnt].next=head[a];
head[a]=cnt++;
}
//P[i][j]表示从i这个点出发 异或值为j的情况总数
void all(int a,int b){
if(P[a].size()<P[b].size()){
swap(P[a],P[b]);
}
map<int,int>::iterator it;
while(P[b].size()>0){
it=P[b].begin();
P[a][it->first]+=it->second;
P[b].erase(it);
}
}
void dfs(int u,int fa,int val){
P[u][val]++;
for(int i=head[u];i!=-1;i=e[i].next){
int v=e[i].u;
if(v==fa)continue;
dfs(v,u,val^v);
map<int,int>::iterator it;
for(it=P[v].begin();it!=P[v].end();it++){
int t=it->first^u;
if(P[u].count(t)){
ans+=(ll)P[u][t]*it->second;
}
}
all(u,v);
}
}
int main(){
scanf("%d",&n);
int a,b;
memset(head,-1,sizeof(head));
for(int i=1;i<n;i++){
scanf("%d%d",&a,&b);
adde(a,b);
adde(b,a);
}
dfs(1,0,1);
printf("%lld",ans);
return 0;
}