树的同构 Java

最新推荐文章于 2024-10-01 18:51:18 发布

DMDMCAR

最新推荐文章于 2024-10-01 18:51:18 发布

阅读量102

点赞数

分类专栏：数据结构文章标签： java

本文链接：https://blog.csdn.net/ZDQQQqq/article/details/124147627

版权

数据结构专栏收录该内容

26 篇文章 0 订阅

订阅专栏

编程语言：Java
题目：
在这里插入图片描述
题解：终于写出来了，成就感满满，解释见注解。
结果：AC

import java.io.*;
import java.util.Scanner;

public class Main {
    static StreamTokenizer in = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
    static PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
    static Scanner sc = new Scanner(new BufferedInputStream(System.in));

    static node[] a,b;

    public static void main(String[] args) throws IOException {
        int n=sc.nextInt();
        a=new node[n];
        sc.nextLine();
        for(int i=0;i<n;i++){
            String[] str=sc.nextLine().split(" ");
            a[i]=new node(str[0],str[1],str[2]);
        }
        int m=sc.nextInt();
        b=new node[m];
        sc.nextLine();
        for(int i=0;i<m;i++){
            String[] str=sc.nextLine().split(" ");
            b[i]=new node(str[0],str[1],str[2]);
        }
        if(n!=m)
      		//两者长度不一致，肯定No
            out.println("No");
        else if(n==0)
        	//两者均为空树，肯定Yes
            out.println("Yes");
        else{
            int x=find(a,"A");
            if(x<0)
            	//第一棵树中没有根结点，肯定No
                out.println("No");
            else{
                int y=find(b,a[x].data);
                if(y<0)
                	//第二棵树中没有相同结点，肯定No
                    out.println("No");
                else{
                	//两棵树都找好根结点了，开始遍历
                    if(search(a[x],b[y]))
                        out.println("Yes");
                    else
                        out.println("No");
                }
            }
        }
        out.flush();
    }

	//在传入数组中，找到数据为data的结点的下标并返回
    private static int find(node[] arr, String data) {
        for(int i=0;i<arr.length;i++){
            if(arr[i].data.equals(data))
                return i;
        }
        return -1;
    }

    private static  boolean search(node x,node y){
    	//写的很通俗易懂，自己看吧
        if(x.data.equals(y.data)){
            boolean flag1=false;
            boolean flag2=false;
            if(x.left.equals("-")){
                if(y.right.equals("-")){
                    if(x.right.equals("-")&&y.left.equals("-"))
                        return true;
                    else if(!x.right.equals("-")&&!y.left.equals("-"))
                        return search(a[Integer.parseInt(x.right)],b[Integer.parseInt(y.left)]);
                }else if(y.left.equals("-")){
                    if(!x.right.equals("-"))
                        return search(a[Integer.parseInt(x.right)],b[Integer.parseInt(y.right)]);
                }else{
                    return false;
                }
            }else{
                if(y.right.equals("-")){
                    if(!x.right.equals("-"))
                        return false;
                    else
                        return search(a[Integer.parseInt(x.left)],b[Integer.parseInt(y.left)]);
                }else if(y.left.equals("-")) {
                    if(!x.right.equals("-"))
                        return false;
                    else
                        return search(a[Integer.parseInt(x.left)],b[Integer.parseInt(y.right)]);
                }else{
                    flag1=search(a[Integer.parseInt(x.left)],b[Integer.parseInt(y.right)])&&search(a[Integer.parseInt(x.right)],b[Integer.parseInt(y.left)]);
                    flag2=search(a[Integer.parseInt(x.left)],b[Integer.parseInt(y.left)])&&search(a[Integer.parseInt(x.right)],b[Integer.parseInt(y.right)]);
                    return flag1||flag2;
                }
            }
        }
        return false;
    }
}

class node{
    String data;
    String left;
    String right;

    public node(String data, String left, String right) {
        this.data=data;
        this.left = left;
        this.right = right;
    }
}