数据结构与算法——03-树1 树的同构（浙大陈姥姥《数据结构与算法》）

1. 题目描述

给定两棵树T1和T2。如果T1可以通过若干次左右孩子互换就变成T2，则我们称两棵树是“同构”的。例如图1给出的两棵树就是同构的，因为我们把其中一棵树的结点A、B、G的左右孩子互换后，就得到另外一棵树。而图2就不是同构的。

现给定两棵树，请你判断它们是否是同构的。

 

输入格式:

输入给出2棵二叉树树的信息。对于每棵树，首先在一行中给出一个非负整数N (≤10)，即该树的结点数（此时假设结点从0到N−1编号）；随后N行，第i行对应编号第i个结点，给出该结点中存储的1个英文大写字母、其左孩子结点的编号、右孩子结点的编号。如果孩子结点为空，则在相应位置上给出“-”。给出的数据间用一个空格分隔。注意：题目保证每个结点中存储的字母是不同的。

输出格式:

如果两棵树是同构的，输出“Yes”，否则输出“No”。

输入样例1（对应图1）：

8
A 1 2
B 3 4
C 5 -
D - -
E 6 -
G 7 -
F - -
H - -
8
G - 4
B 7 6
F - -
A 5 1
H - -
C 0 -
D - -
E 2 -

输出样例1:

Yes

输入样例2（对应图2）：

8
B 5 7
F - -
A 0 3
C 6 -
H - -
D - -
G 4 -
E 1 -
8
D 6 -
B 5 -
E - -
H - -
C 0 2
G - 3
F - -
A 1 4

输出样例2:

No

2. 题目理解

编程主要分为两块：

1. 读取输入构成二叉树；
2. 判断两个二叉树是否同构。

构成二叉树的重点是如何找到二叉树的根节点，这个之前的博文已经有介绍，主要是通过该节点是否在输入中出现过，如果出现过，说明该节点是某个节点的子节点，那么输入中没有出现过的那个节点就是根节点。

输入树的根节点，判断两个树是否同构的逻辑（按照顺序往下，通过了上面的判断才能到达下面的判断逻辑）：

1. 输入节点为空（NULL）：同构，return1（空也认为是相等的一种特例）；

2. 输入节点一个为空，另一个不为空：不同构，return0；

3. 输入节点都不为空，但是两个节点值不相等：不同构，return0；

4. 输入两个节点的左儿子存在并且相等，那么return（递归调用两个左儿子）&&（递归调用两个右儿子）；

5. 到这一步，说明两个节点的左儿子存在但不相等，因此return（递归调用节点1的左儿子和节点2的右儿子）&&（递归调用节点1的右儿子和节点2的左儿子）；

有了这个理解就不难写出判断两个书是否同构的函数了。

3. C++代码

#include <iostream>
#include <cctype>
#include <vector>
#include <queue>

using namespace std;
typedef struct TreeNode {//二叉树Node定义
	char val;
	int left;
	int right;
}Node;
int ReadInput(vector<Node>&nodeList);//循环读取输入数据
int Isomorphic(vector<Node>&nodeList1, int rootIdx1, vector<Node>&nodeList2, int rootIdx2);//判断两个树是否同构
int main()
{
	int num;
	cin >> num;//读入数据行数n
	vector<Node> nodeList1(num);//创建节点数组1
	//循环读入输入节点数据，并找到树1根节点的位置
	int rootIdx1 = ReadInput(nodeList1);
	cin >> num;//读入数据行数n
	vector<Node> nodeList2(num);//创建节点数组2
	//循环读入输入节点数据，并找到树2的根节点的位置
	int rootIdx2 = ReadInput(nodeList2);
	if(Isomorphic(nodeList1, rootIdx1, nodeList2, rootIdx2))cout<<"Yes"<<endl;
	else cout << "No" << endl;
	return 0;
}
//判断两个树是否同构
int Isomorphic(vector<Node>&nodeList1, int idx1, vector<Node>&nodeList2, int idx2) {
	if (nodeList1.size() == 0 && nodeList2.size() == 0)return 1;
	if (idx1 == -1 && idx2 == -1)return 1;//如果当前节点不存在（为NULL）返回1
	if ((idx1 == -1 && idx2 != -1) || (idx1 != -1 && idx2 == -1))return 0;//一个存在一个不存在，返回0
	if (nodeList1[idx1].val != nodeList2[idx2].val)return 0;//当前节点值不相等，返回0
	if (nodeList1[idx1].left == -1 && nodeList2[idx2].left == -1)//当前节点相等，并且当前节点不存在左子节点，递归调用右儿子
		return Isomorphic(nodeList1, nodeList1[idx1].right, nodeList2, nodeList2[idx2].right);
	if ((nodeList1[idx1].left != -1 && nodeList2[idx2].left != -1) &&
		(nodeList1[nodeList1[idx1].left].val == nodeList2[nodeList2[idx2].left].val))//左儿子存在且相等，递归调用左儿子和右儿子
		return Isomorphic(nodeList1, nodeList1[idx1].left, nodeList2, nodeList2[idx2].left) &&
		Isomorphic(nodeList1, nodeList1[idx1].right, nodeList2, nodeList2[idx2].right);
	else return Isomorphic(nodeList1, nodeList1[idx1].left, nodeList2, nodeList2[idx2].right)&&//左儿子存在但是不相等，递归调用左右，右左
		Isomorphic(nodeList1, nodeList1[idx1].right, nodeList2, nodeList2[idx2].left);
}

//循环读取输入数据
int ReadInput(vector<Node>&nodeList) {
	int num = nodeList.size();
	
	//list用来记录根节点，由于根节点不是任何节点的子节点，因此初始化list为1，
	//将每个节点的左右节点idx的对应list[idx]设为-1，表示该节点不是根节点；最后剩下唯一的不为空的list[idx]，其idx就是根节点的val
	vector<int> list(num, 1);
	char left, right, val;
	for (int i = 0; i < num; i++) {	//循环读入
		cin >> val >> left >> right;
		nodeList[i].val = val;
		//判断当前节点的左右节点是否为空，如果为空，将子节点idx设为-1
		if (left != '-') {
			nodeList[i].left = left - '0';
			list[nodeList[i].left] = 0;
		}
		else nodeList[i].left = -1;

		if (right != '-') {
			nodeList[i].right = right - '0';
			list[nodeList[i].right] = 0;
		}
		else nodeList[i].right = -1;
	}
	for (int i = 0; i < num; i++) {
		if (list[i])return i;//i是根节点所在的位置
	}
}