Aizu - 2784 Similarity of Subtrees JAG2016

https://vjudge.net/contest/145955#problem/E

一开始想了半天怎么匹配。。。

后来发现可以直接对这个值做一个hash，就是个水题了。。。因为每层的个数不超过n，所以bas设个比较大的质数就行了

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;

const int maxl=3e5+10;
const int bas=1e9+9;

int n,m,k,tot,cnt;ll ans;
int a[maxl];ull val[maxl];
vector<int> e[maxl];
char s[maxl];
bool vis[maxl];
map<ull,int> mp;

inline void prework()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		e[u].push_back(v);
	}
}

inline void dfs(int u)
{
	val[u]=1;
	for(int v:e[u])
	{
		dfs(v);
		val[u]+=val[v]*bas;
	}
	ans+=mp[val[u]];
	mp[val[u]]++;
}

inline void mainwork()
{
	dfs(1);
}

inline void print()
{
	printf("%lld\n",ans);
}

int main()
{
	int t=1;
	//scanf("%d",&t);
	while(t--)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}