【算法】劍指 Offer 26. 樹的子結構 和 面試題 04.10. 檢查子樹

輸入兩棵二叉樹A和B,判斷B是不是A的子結構。(約定空樹不是任意一個樹的子結構)

B是A的子結構, 即 A中有出現和B相同的結構和節點值。

例如:
給定的樹 A:

     3
    / \
   4   5
  / \
 1   2
給定的樹 B:

   4 
  /
 1

代碼:

func isSubStructure(A *TreeNode, B *TreeNode) bool {
	return (A != nil && B != nil) && (isSubStructure(A.Left, B) || isSubStructure(A.Right, B) || isSubStructureHelp(A, B))
}

func isSubStructureHelp(A *TreeNode, B *TreeNode) bool {
	if B == nil {
		return true
	}
	if A == nil {
		return false
	}
	return (A.Val == B.Val) && isSubStructureHelp(A.Right, B.Right) && isSubStructureHelp(A.Left, B.Left)
}

檢查子樹。你有兩棵非常大的二叉樹:T1,有幾萬個節點;T2,有幾萬個節點。設計一個算法,判斷 T2 是否爲 T1 的子樹。

如果 T1 有這麼一個節點 n,其子樹與 T2 一模一樣,則 T2 爲 T1 的子樹,也就是說,從節點 n 處把樹砍斷,得到的樹與 T2 完全相同。

示例1:

 輸入:t1 = [1, 2, 3], t2 = [2]
 輸出:true
示例2:

 輸入:t1 = [1, null, 2, 4], t2 = [3, 2]
 輸出:false
提示:

樹的節點數目範圍爲[0, 20000]。

來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/check-subtree-lcci
著作權歸領釦網絡所有。商業轉載請聯繫官方授權,非商業轉載請註明出處。

代碼:

func checkSubTree(t1 *TreeNode, t2 *TreeNode) bool {
	if t1 == nil {
		return false
	}
	return isSubtreeHelp(t1, t2) || checkSubTree(t1.Right, t2) || checkSubTree(t1.Left, t2)
}

func isSubtreeHelp(s, t *TreeNode) bool {
	if s == nil && t == nil {
		return true
	}
	if s == nil || t == nil {
		return false
	}
	if s.Val == t.Val {
		return isSubtreeHelp(s.Right, t.Right) && isSubtreeHelp(s.Left, t.Left)
	}
	return false
}

*注意兩道題目的區別,子樹和子結構不一樣

 

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