题目描述
输入两棵二叉树A,B,判断 B 是不是 A 的子结构。(ps:我们约定空树不是任意一个树的子结构)
https://www.nowcoder.com/practice/6e196c44c7004d15b1610b9afca8bd88
解题想法
找到相同的根节点,遍历左右子树,相同返回 true
遍历大树,找到两个相同值的根节点
对相同的节点子树进行遍历,若有不同的返回 false。
若 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);
}
}