114. Flatten Binary Tree to Linked List
題目:
https://leetcode.com/problems/flatten-binary-tree-to-linked-list/
難度:
Medium
有hint,又是玩DFS的preorder,用的loop,也不算很聰明的算法
用stack來(lái)放node,每次把pop出來(lái)的node左邊變成null,把它的right指向下一個(gè)pop出來(lái)的node,也就是為嘛有prev,然后root處特殊處理一下
class Solution(object):
def flatten(self, root):
"""
:type root: TreeNode
:rtype: void Do not return anything, modify root in-place instead.
"""
if root == None:
return
s = []
s.append(root)
prev = root
while s:
node = s.pop()
if node.right:
s.append(node.right)
if node.left:
s.append(node.left)
node.left = None
if node == root:
continue
else:
prev.right = node
prev = node