苦恼的郭大侠
题目连接:
http://acm.uestc.edu.cn/#/problem/show/1284
Description
花开雷霆崖,血染伊吕波。
公元1772年。
郭大侠终于照着天行廖的叮嘱,摆出了阵型,准备带着部队与天行廖相逢于雷霆崖。
但郭大侠实在是没有把握,这个阵型是真的如天行廖所言,所向披靡,于是准备在于天行廖相逢前再确认一遍。
按照天行廖的叮嘱说,这个阵型得满足以下两个要求才能发挥最大威力:
1.这个阵型是一颗树,即n个点,n-1条边,任意两点之间都能互相到达。
2.这个阵型满足好朋友的对数大于n对,好朋友即满足P(u,v)=0,P(u,v)表示从u点到v点的简单路径上点编号的异或和。注意P(u,v)和P(v,u)视作一样,只计数一次。
郭大侠的士兵的的确确摆出了一颗树,但是点数太多了,很难统计好朋友的对数。
这可怎么办?
现在郭大侠找到了你,希望你能帮助憔悴的郭大侠解决这个问题,希望你能够数出这棵树上究竟有多少对好朋友。
Input
第一行n,表示点的数量,(1<=n<=100000)
接下来n-1行,每行两个整数u,v,表示点u与点v之间连有一条边(1<=u,v<=n)
保证是一棵树
Output
输出这棵树上究竟有多少对好朋友。
Sample Input
17
9 8
9 11
9 13
9 15
9 16
11 10
13 12
15 14
16 17
8 1
1 5
5 4
4 6
6 7
7 2
2 3
Sample Output
18
Hint
题意
题解:
map启发式合并就好了
对于每个点暴力统计答案就好了
代码
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+7;
vector<int> E[maxn];
map<int,int> H[maxn];
int pre[maxn];
long long ans = 0;
void uni(int x,int y)
{
if(H[x].size()<H[y].size())
swap(H[x],H[y]);
map<int,int>::iterator it;
while(H[y].size())
{
it=H[y].begin();
H[x][it->first]+=it->second;
H[y].erase(it);
}
}
void dfs(int x,int fa,int val)
{
H[x][val]++;
for(int i=0;i<E[x].size();i++)
{
int v = E[x][i];
if(v == fa)continue;
dfs(v,x,val^v);
for(auto a:H[v])
{
int t = a.first^x;
if(H[x].count(t))
ans+=1ll*H[x][t]*a.second;
}
uni(x,v);
}
}
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<n;i++)
{
int x,y;scanf("%d%d",&x,&y);
E[x].push_back(y);
E[y].push_back(x);
}
dfs(1,0,1);
cout<<ans<<endl;
}