题目要求
树的子结构
- 输入两棵二叉树A,B,判断B是不是A的子结构。
- (ps:我们约定空树不是任意一个树的子结构)
核心思想
递归思想。
完整代码如下
/**
* 树的子结构
* 输入两棵二叉树A,B,判断B是不是A的子结构。
* (ps:我们约定空树不是任意一个树的子结构)
*
*/
public class Solution {
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode (int val) {
this.val = val;
}
}
public boolean hasSubtree(TreeNode root1, TreeNode root2) {
if((root1 == null) || (root2 == null)) {
return false;
}
boolean flag = false;
if(root1.val == root2.val) {
flag = doHasSubtree(root1,root2);
}
if(flag)
return flag;
if(!flag) {
flag = hasSubtree(root1.left, root2);
if(flag)
return flag;
else {
flag = hasSubtree(root1.right, root2);
if(flag)
return flag;
}
}
return false;
}
public boolean doHasSubtree(TreeNode root1, TreeNode root2) {
if(root2 == null)
return true;
if(root1 == null)
return false;
if(root1.val != root2.val) {
return false;
}
return doHasSubtree(root1.left, root2.left) && doHasSubtree(root1.right, root2.right);
}
}