leetcode 987

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