树的子结构【剑指offer——JAVA实现】

【考点：递归和树的遍历】树的子结构

题目描述

输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）

解题思路

本题我有两种解题思路：

1、使用后序遍历二叉树A/B，得到后序遍历数组后判断两者是否包含的关系

2、使用递归去比较每个元素

代码实现1

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;
    }
}
*/

import java.util.ArrayList;
public class Solution {
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        if(root1 == null || root2 == null) return false;
        //方法1：拿到root1与root2的后序遍历
        ArrayList<Integer> list1 = new ArrayList<Integer>();
        ArrayList<Integer> list2 = new ArrayList<Integer>();
        
        getPostOrder(root1, list1);
        getPostOrder(root2, list2);
        //判断两个树的后序遍历是否存在包含关系，有则说明是其子结构
        if(list1.containsAll(list2))
            return true;
    }

    private void getPostOrder(TreeNode root, ArrayList<Integer> list) {
        if(root == null) return;
        getPostOrder(root.left, list);
        getPostOrder(root.right, list);
        list.add(root.val);
    }
}

代码实现2

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/

import java.util.ArrayList;
public class Solution {
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        if(root1 == null || root2 == null) return false;

        //方法2：递归检测
        boolean result = false;
        if(root1.val == root2.val) {
            result = check(root1, root2);
        }
        
        if(!result) {
            result = HasSubtree(root1.right, root2);
        }
        
        if(!result) {
            result = HasSubtree(root1.left, root2);
        }
        
        return result;
    }
    
    private boolean check(TreeNode node1,TreeNode node2) {
        //如果root2都检查完了，都匹配，则返回true
        if(node2 == null) return true;
        //如果root1都检查完了，不匹配，则返回false
        if(node1 == null) return false;
        
        if(node1.val != node2.val) return false;
        
        return check(node1.left, node2.left) && check(node1.right, node2.right);
    }

}

更多算法解答请点击《剑指offer》66题JAVA代码算法实现全集