ZOJ 3602 Count the Trees

开始以为子树不仅需要同构，还要每个节点的编号相等，然后发现测试样例怎么对不上的呢。于是看了解题报告，于是思路就暴露了。一个节点对应一颗子树，所以可以为每个节点映射一个整数，这个整数唯一体现了这颗子树的形态。显然这个整数可以递归地由两个孩子对应的整数确定，方便起见，我们把第一次碰到的子树（一个节点）编号为0，然后第二次遇到的编号为1，依次类推。由于写了两个类似的的DFS，调用的时候掉混了，这是第二次了吧。

#include <cstdio>
#include <map>
#include <memory.h>
using namespace std;
#define pack(a,b) make_pair(a,b)

map<pair<int, int>, int>::iterator it;
map<pair<int, int>, int> kind;
int left[100001], right[100001];
int count[100000];
int n, m, num;
long long ans;

inline int match(pair<int, int> p)
{
	it = kind.find(p);
	if(it == kind.end())
		return (kind[p] = num++);
	else
		return it->second;
}

int dfs(int ind)
{
	if(ind == -1) return ind;
	left[ind] = dfs(left[ind]);
	right[ind] = dfs(right[ind]);
	left[ind] = match(pack(left[ind], right[ind]));
	++ count[left[ind]];
	return left[ind];
}

int cal(int ind)
{
	if(ind == -1) return -1;
	left[ind] = cal(left[ind]); //dfs-->cal
	right[ind] = cal(right[ind]);
	left[ind] = match(pack(left[ind], right[ind]));
	if(left[ind] < 100000)
		ans += count[left[ind]];
	return left[ind];
}

int main()
{
	int T;
	scanf("%d", &T);
	while(T --)
	{
		scanf("%d%d", &n, &m);
		for(int i=1; i<=n; ++i)
			scanf("%d%d", &left[i], &right[i]);
		num=0, kind.clear();
		memset(count, 0, sizeof(count));
		dfs(1);
		for(int i=1; i<=m; ++i)
			scanf("%d%d", &left[i], &right[i]);
		ans=0, cal(1);
		printf("%lld\n", ans);
	}
	return 0;
}