树的同构 Java 中国大学MOOC-陈越、何钦铭-数据结构-2020秋

树的同构 Java 中国大学MOOC-陈越、何钦铭-数据结构-2020秋

给定两棵树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

下面为解答代码：

import java.util.Scanner;
//判断树的同构，首先要读入两棵树
//再判断是否同构
public class Main {
 public static int n1 = -1;
 
 public static void main(String[] args) {
  // TODO Auto-generated method stub
  Scanner in = new Scanner(System.in);
  Node[] nodeA = read(in);
  int root1 = n1;
  
  Node[] nodeB = read(in);
  int root2 = n1;
  
  boolean bool = IsIsomorphism(root1, nodeA, root2, nodeB);
  
  if(bool)
   System.out.println("Yes");
  else
   System.out.println("No");
 }
  
 public static boolean IsIsomorphism(int root1, Node[] A, int root2, Node[] B) {
  //情况1：两个树都为空
  if(root1 == -1 && root2 == -1)
   return true;
  //情况2：一个树有，一个树为空树，那么肯定不同构
  else if((root1 == -1 && root2 != -1) || (root1 != -1 && root2 == -1))  
   return false;
  //情况3：两棵树的树根不等
  else if(A[root1].element != B[root2].element)
   return false;
  //情况4：A 的左边与B左边同构，A右边与B右边同构
  else if(IsIsomorphism(A[root1].left,A,B[root2].left,B) && IsIsomorphism(A[root1].right,A,B[root2].right,B))
   return true;
  //情况5：A 的左边与B的右边同构，A的右边与B的左边同构
  else if(IsIsomorphism(A[root1].left,A,B[root2].right,B) && IsIsomorphism(A[root1].left,A,B[root2].right,B))
   return true;
  
  return false;
 }
  //读入树，用数组来表示二叉树，此处用静态链表来表示二叉树 -1表示没有左儿子或者右儿子
  public static Node [] read (Scanner in) {
  //先读入节点的个数
  int n = in.nextInt();
  //再依次读入各节点的值和其左右节点的位置
  //初始化对象数组
  Node[] node = new Node[n];
  //定义一个check数组来判断根节点为哪一个,初始化check数组，更新check数组的值，没有被更新过的就为根节点
  int[] check = new int[n];
  for(int i=0 ;i<n ;i++) {
   check[i] = 0;
  }
    for(int i=0; i<n; i++) {
   node[i] = new Node();
   
   //先读入一个String类型的节点本身的信息
   String str = in.next();
   //element是一个char型的数据，而读入数据是一个string类型的所以转换一下。
   node[i].element = str.charAt(0);
   
   //读入第二个数据,第二个数据可能是字符类型的‘-’，也可能是int类型的数据，所以先以字符型读入再按照情况转数据
   str = in.next();
   if(!str.equals("-")) {
    int temp;
    temp = Integer.parseInt(str);
    node[i].left = temp;
    check[temp] = 1;//更新check数组的值，被指向的节点设置为1
   }else
    node[i].left = -1;
   
   //读取第三个数据，同上
      str = in.next();
   if(!str.equals("-")) {
    int temp = Integer.parseInt(str);
    node[i].right = temp;
    check[temp] = 1;
   }else
    node[i].right = -1;
  }
  
  for(int i=0; i<n; i++) {
   if(check[i] == 0) {
    //n1为一个全局变量
    n1 = i;
    break;
   }
  }
  
  return node;
 }
}
class Node {
 public char element; //该节点本身的信息
 public int left; //左儿子的位置
 public int right; //右儿子的位置
 
 public Node() {
  this.element = '0';
  this.left = -1;
  this.right = -1;
 }
 public Node(char ele, int le, int ri) {
  this.element = ele;
  this.left = le;
  this.right = ri;
 }
}