Leetcode 951:翻轉等價二叉樹(超詳細的解法!!!)
阿新 • • 發佈:2018-12-09
我們可以為二叉樹 T 定義一個翻轉操作,如下所示:選擇任意節點,然後交換它的左子樹和右子樹。
只要經過一定次數的翻轉操作後,能使 X 等於 Y,我們就稱二叉樹 X 翻轉等價於二叉樹 Y。
編寫一個判斷兩個二叉樹是否是翻轉等價的函式。這些樹由根節點 root1
和 root2
給出。
示例:
輸入:root1 = [1,2,3,4,5,6,null,null,null,7,8], root2 = [1,3,2,null,6,4,5,null,null,null,null,8,7] 輸出:true 解釋:We flipped at nodes with values 1, 3, and 5.
提示:
- 每棵樹最多有
100
個節點。 - 每棵樹中的每個值都是唯一的、在
[0, 99]
範圍內的整數。
解題思路
這個問題很明顯通過遞迴來解決。我們定義函式 表示以 和 為根的結果,那麼 。我們接著考慮邊界問題,我們判斷 ,如果成立的話,我們要返回 。我們還要判斷 ,如果成立的話,我們返回 。
class Solution:
def flipEquiv(self, root1, root2):
"""
:type root1: TreeNode
:type root2: TreeNode
:rtype: bool
"""
if root1 == None or root2 == None:
return root1 == root2
if root1.val != root2.val:
return False
return (self.flipEquiv(root1.left, root2.left) and self.flipEquiv(root1.right, root2.right)) or\
(self.flipEquiv(root1.left, root2.right) and self.flipEquiv(root1.right, root2.left))
我將該問題的其他語言版本新增到了我的GitHub Leetcode
如有問題,希望大家指出!!!