题目
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
例如:
给定的树 A:
3
/ \
4 5
/ \
1 2
给定的树 B:
4
/
1
返回 true,因为 B 与 A 的一个子树拥有相同的结构和节点值。
解题思路
采用递归
1、先判断是B是否是A以其根节点的一个子结构
若不成立,则判断B是否是A左/右子树的一个子结构
2、构造函数isssubTree(TA,TB)
TB是否是以TA为根节点的子树
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public boolean isSubStructure(TreeNode A, TreeNode B) {
if(A==null||B==null){
return false;
}
boolean[] flag=new boolean [2];
Pre(A,B,flag);
return flag[0];
}
public void Pre(TreeNode A,TreeNode B,boolean[] flag ){
if(A!=null){
if(isSubTree(A,B)){
flag[0]=true;
}
Pre(A.left,B,flag);
Pre(A.right,B,flag);
}
}
public boolean isSubTree(TreeNode Ta,TreeNode Tb){
if(Tb==null){
return true;
}
if(Ta==null){
return false;
}
if(Ta.val!=Tb.val){
return false;
}
return isSubTree(Ta.left,Tb.left)&&isSubTree(Ta.right,Tb.right);
}
}