題目:
輸入兩棵二叉樹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;
}
}