树的子结构

题目描述

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

https://www.nowcoder.com/practice/6e196c44c7004d15b1610b9afca8bd88

解题想法

找到相同的根节点，遍历左右子树，相同返回 true

1. 遍历大树，找到两个相同值的根节点

2. 对相同的节点子树进行遍历，若有不同的返回 false。

3. 若 B 为 null，说明子树遍历完成都相同，返回 true。若 A 为 null，说明父树遍历完成还没有找到相同的，返回 false。

要注意的是，对子节点进行遍历时，应该先判断 B 是否为 null，再判断 A 是否为 null。因为 A 节点为 null 时候，B 节点也可能为 null，但 B 是 A 的子结构。

代码实现

/**
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) {
    	// 题目中标明 null 不为任一子结构
        if (root1 == null || root2 == null){
            return false;
        }
        // 当前节点相同，那么继续遍历子节点
        if (root1.val == root2.val){
            if (isOK(root1,root2)){
                return true;
            }
        }
        // 在父树的子树中找到值相同的节点，然后进行判断
        return HasSubtree(root1.left,root2) || HasSubtree(root1.right,root2);
    }
    
    // 对两个值相同的节点，进行遍历，判断 root2 是否是子结构
    public boolean isOK(TreeNode root1,TreeNode root2){
    	// 必须先判断 root2，当 root2 为 null 时，就不用管 root1 的值
        if (root2 == null){
            return true;
        }
        // 此时 root2 != null 而 root1 == null，那么返回 false
        if (root1 == null){
            return false;
        }
        // 节点值有不一样的
        if (root1.val != root2.val){
            return false;
        }
        // 遍历左右子树，不一样返回 false，直到 root2 == null
        return isOK(root1.left,root2.left) && isOK(root1.right,root2.right);
    }
    
}