题目:输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
思路:将原始问题分为两个子问题
a. 在树A中寻找和树B根节点的值一样的结点R
b. 判断在树A中以R为根节点的子树,是否和树B拥有一样的结构
树结构:
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
核心代码:
public class Solution {
public boolean doesTree1HasTree2(TreeNode root1, TreeNode root2){
//将Tree2都遍历完了,都能对应上,返回true
if(root2 == null){
return true;
}
//将子树Tree1遍历完,仍然没有对应上,返回false
if(root1 == null){
return false;
}
//如果两树当前节点值不相等,直接返回false
if(root1.val != root2.val){
return false;
}
//如果根节点对的上,再去找左右子节点是否匹配
return doesTree1HasTree2(root1.left, root2.left) && doesTree1HasTree2(root1.right, root2.right);
}
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
boolean result = false;
//如果Tree1和Tree2有一个为空,直接返回false
if(root1 !=null && root2 != null){
//如果找到对应Tree2的根节点,
if(root1.val == root2.val){
//以Tree2根节点为起点,看Tree1中是否包含Tree2
result = doesTree1HasTree2(root1, root2);
}
//如果根节点对应不上,去看Tree1的左节点是否能和Tree2对上
if(!result){
result = HasSubtree(root1.left, root2);
}
//如果还找不到,再去看Tree1的右节点是否能和Tree2对上
if(!result){
result = HasSubtree(root1.right, root2);
}
}
return result;
}
}