一口气带你刷5道题
一套代码，（几乎）改都不用改，刷完各种树的各种遍历。
先说原理：一个栈，栈中每个元素是由一个元组组成，根据进入栈中树的节点需要扮演角色不同，元祖第二个布尔值不同。有个题解讲得很棒。
原理明白后只需搞懂前序，后面都很简单！
对于二叉树中的任何一个节点而言，它都有两个角色需要扮演，一个是作为值存储的角色（角色1），另一个角色是作为它所带领的子树的一个代表（角色2）。而我们设置的bool变量，就是为了说明我当前拿到的这个节点，应该是以一个值存储的这种角色对待它(True)，还是应该以一个子树的代表这种角色对待它（False），如果是前者，那么就简单的将其所存储的值打印出来，如果是后者，我们需要继续探索由它带领的子树。
作者：liu-yong-4
链接：https://leetcode-cn.com/problems/two-sum/solution/xian-xu-zhong-xu-hou-xu-de-fei-di-gui-ban-ben-by-l/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
二叉树前序
根据栈的特点，先右后左，最后根。第一次进栈代表整棵子树False，还未被访问，第二次进栈代表节点True，再次出栈就要被访问。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def preorderTraversal(self, root: TreeNode) -> List[int]:
res =[]
stack = [] #must have
if root:
stack = [(root,False)]
while stack:
node,visited= stack.pop()
if visited:
res.append(node.val)
else:
if node.right:
stack.append([node.right,False])
if node.left:
stack.append([node.left,False])
stack.append([node,True]) #前序
return res
二叉树中序
中序和后续只需要改节点进栈时机一句代码。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def preorderTraversal(self, root: TreeNode) -> List[int]:
res =[]
stack = [] #must have
if root:
stack = [(root,False)]
while stack:
node,visited= stack.pop()
if visited:
res.append(node.val)
else:
if node.right:
stack.append([node.right,False])
stack.append([node,True]) #中序
if node.left:
stack.append([node.left,False])
return res
二叉树后序
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def preorderTraversal(self, root: TreeNode) -> List[int]:
res =[]
stack = [] #must have
if root:
stack = [(root,False)]
while stack:
node,visited= stack.pop()
if visited:
res.append(node.val)
else:
stack.append([node,True]) #后序
if node.right:
stack.append([node.right,False])
if node.left:
stack.append([node.left,False])
return res
N叉树前序
只需在二叉树前序基础上，进右左改为逆向进孩子。
"""
# Definition for a Node.
class Node:
def __init__(self, val, children):
self.val = val
self.children = children
"""
class Solution:
def preorder(self, root: 'Node') -> List[int]:
res =[]
stack = [] #must have
if root:
stack = [(root,False)]
while stack:
node,visited= stack.pop()
if visited:
res.append(node.val)
else:
for child in node.children[::-1]: #逆向进孩子。
stack.append([child,False])
stack.append([node,True]) #前序
return res
N叉树后序
比葫芦画瓢，没什么好说。
"""
# Definition for a Node.
class Node:
def __init__(self, val, children):
self.val = val
self.children = children
"""
class Solution:
def preorder(self, root: 'Node') -> List[int]:
res =[]
stack = [] #must have
if root:
stack = [(root,False)]
while stack:
node,visited= stack.pop()
if visited:
res.append(node.val)
else:
stack.append([node,True]) #后序
for child in node.children[::-1]: #逆向进孩子。
stack.append([child,False])
return res
2019.08.15