剑指offer(十七)——树的子结构
题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
题解
首先理解题意,B是A的子结构的前提是,B树是A树的一部分,即从B的根节点开始,要与A其中一个结点开始,结点值完全一致。
- 因为已经说了,空树不是子结构,所以先行判定两树是否为空。
- 然后我们需要在A树中找到与B树根节点对应的节点,从那里开始进行判定,因此我们创建了一个函数IsSubtree(TreeNode,TreeNode)用来判定所有节点是否相符。
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
if (root1 == null || root2 == null) {
return false;
}
//从该节点开始判定是否相符
//从A树的左孩子开始判定是否相符
//从A树的右孩子开始判定是否相符
return IsSubtree(root1, root2) || HasSubtree(root1.left, root2) || HasSubtree(root1.right, root2);
}
public boolean IsSubtree(TreeNode root1, TreeNode root2) {
//当B树遍历完之后,即B树为A树子结构,返回true
if (root2 == null) {
return true;
}
//当A树遍历完,但是B树没有,证明不是子结构,返回false
if (root1 == null) {
return false;
}
//当该节点一致时,同时向其孩子搜索判定
if (root1.val == root2.val) {
return IsSubtree(root1.left, root2.left) && IsSubtree(root1.right, root2.right);
}else {
return false;
}
}