25.树的子结构
输入两棵二叉树 A,B,判断 B 是不是 A 的子结构。
我们规定空树不是任何树的子结构。
数据范围:
每棵树的节点数量 [0,1000]。
样例:
树A:
树B:
返回true,因为B是A的子结构
代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public boolean hasSubtree(TreeNode pRoot1, TreeNode pRoot2) {
//边界条件处理:空树不是任何树的子结构
if(pRoot1 == null||pRoot2 == null)return false;
//递归检查三种情况:
//1.以A为根节点的子树包含B
//2.B在A的左子树中
//3.B在A的右子树中
return dfs(pRoot1,pRoot2)||hasSubtree(pRoot1.left,pRoot2)||hasSubtree(pRoot1.right,pRoot2);
}
public boolean dfs(TreeNode A,TreeNode B){
//1.如果B已经遍历完毕,说明匹配成功
if(B == null)return true;
//2.如果A已经遍历完但B还有节点,或者节点值不匹配,则失败
if(A == null||A.val != B.val)return false;
//3.递归检查左右子树
return dfs(A.left,B.left)&&dfs(A.right,B.right);
}
}