劍指Offer之二叉樹的子結構

題目:

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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章