【树】剑指Offer: 树的子结构

【在线编程】树的子结构

【问题描述】

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

【解题思路 & Java实现】

思路：首先在 root1中找到与root2的根结点相等的结点（称为候选结点，有可能从该结点开始匹配到root2), 然后从该候选结点开始进行匹配root2。如果匹配成功，则返回true；如果匹配失败，则在root1的左子树中寻找下一个候选结点并进行匹配；如果左子树中仍匹配失败，则在root1的右子树中寻找候选结点并进行匹配。

/**
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) {
        if (root1 == null || root2 == null) {
            return false;
        }

        boolean flag = false;
        if (root1.val == root2.val) {
            flag = matchSubTree(root1, root2);
        }
        //候选结点匹配失败，则在左子树中寻找候选结点并进行匹配
        if (!flag) {
            flag = HasSubtree(root1.left, root2);
        }

        //左子树中没有找到候选结点或者候选结点匹配失败，则在右子树中寻找候选结点并进行匹配
        if (!flag) {
            flag = HasSubtree(root1.right, root2);
        }

        return flag;
    }

    //对候选结点进行匹配
    public boolean matchSubTree(TreeNode root1, TreeNode root2) {
        //递归出口一：root2中的结点全部匹配成功
        if (root2 == null) {
            return true;
        }
        //递归出口二：root2 还没遍历完，root1已经遍历完了
        if (root1 == null) {
            return false;
        }
        //递归出口三：匹配过程中，某个结点匹配不成功
        if (root1.val != root2.val) {
            return false;
        }

        //递归匹配左子树和右子树
        return matchSubTree(root1.left, root2.left) && matchSubTree(root1.right, root2.right);
    }
}