[Aizu2784]Similarity of Subtrees

vjudge

Description

给一棵\(n\)个节点的有根树，定义两棵树同构当且仅当他们每个深度的节点个数相同。问这个树上有多少对子树满足同构。\(n\le100000\)。

sol

树\(hash\)。

每个深度的节点个数，类似于一个多项式？所以定义\(hash\)函数:

\[Hash(u)=base1+base2\sum Hash(v)
\]

最后判相等就不说了，\(sort\)一遍再搞一搞就好了。

code

在我交换\(base1\)和\(base2\)的值之前它是\(WA\)的。。。

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int gi(){
	int x=0,w=1;char ch=getchar();
	while ((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
	if (ch=='-') w=0,ch=getchar();
	while (ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
	return w?x:-x;
}
#define ull unsigned long long
const int N = 2e5+5;
const ull base2 = 20020415;
const ull base1 = 20011118;
int n,to[N],nxt[N],head[N],cnt;
ull Hash[N],ans;
void link(int u,int v){
	to[++cnt]=v;nxt[cnt]=head[u];head[u]=cnt;
}
void dfs(int u,int f){
	Hash[u]=base1;
	for (int e=head[u];e;e=nxt[e])
		if (to[e]!=f) dfs(to[e],u),Hash[u]+=Hash[to[e]]*base2;
}
int main(){
	n=gi();
	for (int i=1;i<n;++i){
		int u=gi(),v=gi();
		link(u,v);link(v,u);
	}
	dfs(1,0);sort(Hash+1,Hash+n+1);
	for (int i=1,j;i<=n;i=j){
		j=i+1;
		while (j<=n&&Hash[j]==Hash[i]) ++j;
		ans+=1ll*(j-i)*(j-i-1)/2;
	}
	printf("%llu\n",ans);return 0;
}