题目描述
输入两棵二叉树A
,B
,判断B
是不是A
的子结构。(ps:我们约定空树不是任意一个树的子结构)
解法
递归方式遍历:
- 在树A中找到和树B的根结点值一样的结点R
- 判断树A以R为根结点的子树是否包含与树B一样的结构
/**
* @author bingo
* @since 2018/11/22
*/
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
/**
* 判断root2是否是root1的子树
* @param root1 树1
* @param root2 树2
* @return 是否是子树
*/
public boolean HasSubtree(TreeNode root1, TreeNode root2) {
boolean res = false;
if (root1 != null && root2 != null) {
if (root1.val == root2.val) {
res = isSame(root1, root2);
}
if (!res) {
res = HasSubtree(root1.left, root2);
}
if (!res) {
res = HasSubtree(root1.right, root2);
}
}
return res;
}
private boolean isSame(TreeNode root1, TreeNode root2) {
if (root2 == null) {
return true;
}
if (root1 == null) {
return false;
}
if (root1.val != root2.val) {
return false;
}
return isSame(root1.left, root2.left) && isSame(root1.right, root2.right);
}
}
测试用例
- 功能测试(树A和树B都是普通的二叉树;树B是/不是树A的子结构);
- 特殊输入测试(两棵二叉树的一个或者两个根结点为空指针,二叉树的所有结点都没有左/右子树)。