题目
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
示例 1:
输入:A = [1,2,3], B = [3,1]
输出:false
示例 2:
输入:A = [3,4,5,1,2], B = [4,1]
输出:true
限制:
0 <= 节点个数 <= 10000
思路
分解问题,要判断B树是不是A的子树,那B树的根节点一定是A树的某个节点,我们先遍历A树,找和B树根节点的点,然后拿这个点作为根节点的树去和B树比较,如果B树是该树的子树则直接返回true,否则继续找这样的节点,直到找完还没有则返回false。
代码
for遍历版本(慢于递归版本)
public boolean isSubStructure(TreeNode A, TreeNode B) {
//题目约定:空树不是任何树的子结构
if (B == null || A == null){
return false;
}
//遍历A,找出起点相同的所有点
Stack<TreeNode> stack1 = new Stack<>();
stack1.push(A);
//找值相同的点
while (!stack1.isEmpty()){
TreeNode t = stack1.pop();
if (t.left != null){
stack1.push(t.left);
}
if (t.right != null){
stack1.push(t.right);
}
if (t.val == B.val && doJudgeSubstructure(t,B)){
return true;
}
}
return false;
}
public boolean doJudgeSubstructure(TreeNode A,TreeNode B){
//出口 (刚进去B一定不为空)
if (B == null){
return true;
}
//A被遍历完了,B还没有结束
if (A == null){
return false;
}
if (A.val != B.val){
return false;
}else {
return doJudgeSubstructure(A.left,B.left) && doJudgeSubstructure(A.right,B.right);
}
}
递归版本
public boolean isSubStructure(TreeNode A, TreeNode B) {
return (A != null && B != null) && (doJudgeSubstructure(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B));
}
public boolean doJudgeSubstructure(TreeNode A, TreeNode B) {
if (B == null) {
return true;
}
if (A == null || A.val != B.val) {
return false;
}
return doJudgeSubstructure(A.left, B.left) && doJudgeSubstructure(A.right, B.right);
}