樹4 反轉(zhuǎn)二叉樹

翻轉(zhuǎn)一棵二叉樹。

示例:

輸入:

? ? 4

? /? \

? 2? ? 7

/ \? / \

1? 3 6? 9

輸出:

? ? 4

? /? \

? 7? ? 2

/ \? / \

9? 6 3? 1


第一種思路:

遞歸+交換左右子樹+保留最初的樹

代碼如下:

#?Definition?for?a?binary?tree?node.

#?class?TreeNode:

#?????def?__init__(self,?x):

#?????????self.val?=?x

#?????????self.left?=?None

#?????????self.right?=?None

class?Solution:

????def?invertTree(self,?root:?TreeNode)?->?TreeNode:

????????if?not?root:

????????????return?root

????????tmp0=root

????????tmp=root.right

????????root.right=root.left

????????root.left=tmp

????????self.invertTree(root.right)

????????self.invertTree(root.left)

????????return?tmp0


第二種思路:迭代+深度優(yōu)先,用廣度優(yōu)先搜索的方法來進行解題,將每一層的字?jǐn)?shù)加入到stack中,然后對每一個子樹的左右子樹進行交換

class?Solution:

????def?invertTree(self,?root:?TreeNode)?->?TreeNode:

????????if?not?root?:?return?

????????stack?=?[root]

????????while?stack?:

????????????length?=?len(stack)

????????????for?i?in?range(length):

????????????????cur?=?stack.pop(0)

????????????????cur.left,cur.right?=?cur.right,cur.left

????????????????if?cur.left?:

????????????????????stack.append(cur.left)

????????????????if?cur.right?:

????????????????????stack.append(cur.right)

????????return?root

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容