输入两棵树A,B,判断B是不是A的子结构。

题目描述

输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)

解题方案

本题的思路在于子结构的判断,首先需要判断两棵树的根节点是否相同,若是不同则递归调用其左子树和B树,若仍不同则递归调用其右子树和B树,若都不同则返回false。

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

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

    }

}
*/
public class Solution {
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        boolean result = false;
        if(root2 != null && root1 != null){
            if(root1.val == root2.val){
                //若两个树的根节点一致则只需要判断该树是否相等即可
                result = isEqual(root1,root2);
            }
            if(!result){
                result =  HasSubtree(root1.right,root2);
            }
            if(!result){
                result =  HasSubtree(root1.left,root2);
            }
        }
        return result;
    }
    public boolean isEqual(TreeNode root1,TreeNode root2){
        if(root2 == null){
            return true;
        }
        if(root1 == null){
            return false;
        }
        if(root1.val != root2.val) 
            return false;
        return isEqual(root1.left,root2.left) && isEqual(root1.right,root2.right);

    }
}

版权声明:本文为qq_24325225原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。