987. 二叉树的垂序遍历
给你二叉树的根结点 root ,请你设计算法计算二叉树的 垂序遍历 序列。
对位于 (row, col) 的每个结点而言,其左右子结点分别位于 (row + 1, col - 1) 和 (row + 1, col + 1) 。树的根结点位于 (0, 0) 。
二叉树的 垂序遍历 从最左边的列开始直到最右边的列结束,按列索引每一列上的所有结点,形成一个按出现位置从上到下排序的有序列表。如果同行同列上有多个结点,则按结点的值从小到大进行排序。
返回二叉树的 垂序遍历 序列。
solution
- 使用二维字典:key为了(列号,行号),value:node.val
- 利用先序遍历,将所有节点的信息记录到dic中
- 最后按照列号,遍历一次dic即可
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def verticalTraversal(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
# 使用二维字典保存至,字典的key为坐标(列号,行号)
dicts = defaultdict(dict)
def find(node,row,col):
if node==None:
return
val = node.val
if row not in dicts[col]:
dicts[col][row] =[]
dicts[col][row].append(val)
find(node.left,row+1,col-1)
find(node.right,row+1,col+1)
find(root,0,0)
res=[]
for col in sorted(dicts):
temp =[]
for row in sorted(dicts[col]):
temp+=sorted(dicts[col][row])
res.append(temp)
return res