题目:
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
例如:
给定的树 A:
3
/ \
4 5
/ \
1 2
给定的树 B:
4
/
1
返回 true,因为 B 与 A 的一个子树拥有相同的结构和节点值。
示例 1:
输入:A = [1,2,3], B = [3,1]
输出:false
示例 2:
输入:A = [3,4,5,1,2], B = [4,1]
输出:true
题解:
先在A树中找到B的根节点,然后在一同左右遍历即可,直到B树为null。
public class TreeNode
{
public int val;
public TreeNode left;
public TreeNode right;
public TreeNode(int x) { val = x; }
}
public class Solution
{
//先在A树中找到B的根节点
public bool IsSubStructure(TreeNode A, TreeNode B)
{
if( A == null || B == null)
{
return false;
}
if(A.val == B.val)
{
return isSub(A,B);
}
//递归查找
return IsSubStructure(A.left,B) || IsSubStructure(A.right,B);
}
//一同遍历A树B的左右节点
private bool isSub(TreeNode a,TreeNode b)
{
if(b == null)
{
return true;
}
if( a == null && b != null)
{
return false;
}
//当前A B树节点相同 遍历当前节点左右子节点
if(a.val == b.val)
{
return isSub(a.left,b.left) && isSub(a.right,b.right);
}
return false;
}
}