剑指Offer——树的子结构

By AverageJoeWang
 标签:

树的子结构

  • 题目描述

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

  • 解题思路

    • 首先需要递归Root1树,找到与Root2根一样的节点,这需要一个遍历
    • 找到相同的根节点后,要判断是否子树,仍需要一个一个遍历对比
    • 树的遍历我们一般就用递归来做
  • 代码实现

/**
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 res = false;
        if (root1 != null && root2 != null){
            if (root1.val == root2.val){
                res = DoesHasSubTree(root1, root2);
            }
            if (!res)
                res = HasSubtree(root1.left, root2);
            if (!res)
                res = HasSubtree(root1.right, root2);
        }
        return res;
    }

    public boolean DoesHasSubTree(TreeNode root1,TreeNode root2){
        if(root1 == null && root2 != null) return false;
        if(root2 == null) return true;
        if(root1.val != root2.val) return false;
        return DoesHasSubTree(root1.left, root2.left) && DoesHasSubTree(root1.right, root2.right);
    }
}