【cdoj 1284】郭大侠与苦恼 map的启发式合并

好吧，其实说是启发式合并，但是本题的真正难点还是在于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;
}