987. 二叉树的垂序遍历

Difficulty
Hard
Tags
二叉树
URL
https://leetcode-cn.com/problems/vertical-order-traversal-of-a-binary-tree/
Star
给你二叉树的根结点 root ，请你设计算法计算二叉树的 垂序遍历 序列。
对位于 (row, col) 的每个结点而言，其左右子结点分别位于 (row + 1, col - 1) 和 (row + 1, col + 1) 。树的根结点位于 (0, 0) 。
二叉树的 垂序遍历 从最左边的列开始直到最右边的列结束，按列索引每一列上的所有结点，形成一个按出现位置从上到下排序的有序列表。如果同行同列上有多个结点，则按结点的值从小到大进行排序。
返回二叉树的 垂序遍历 序列。
示例 1：
notion imagenotion image
输入：root = [3,9,20,null,null,15,7]
输出：[[9],[3,15],[20],[7]]
解释：
列 -1 ：只有结点 9 在此列中。
列  0 ：只有结点 3 和 15 在此列中，按从上到下顺序。
列  1 ：只有结点 20 在此列中。
列  2 ：只有结点 7 在此列中。
示例 2：
notion imagenotion image
输入：root = [1,2,3,4,5,6,7]
输出：[[4],[2],[1,5,6],[3],[7]]
解释：
列 -2 ：只有结点 4 在此列中。
列 -1 ：只有结点 2 在此列中。
列  0 ：结点 1 、5 和 6 都在此列中。
          1 在上面，所以它出现在前面。
          5 和 6 位置都是 (2, 0) ，所以按值从小到大排序，5 在 6 的前面。
列  1 ：只有结点 3 在此列中。
列  2 ：只有结点 7 在此列中。
示例 3：
输入：root = [1,2,3,4,6,5,7]
输出：[[4],[2],[1,5,6],[3],[7]]
解释：
这个示例实际上与示例 2 完全相同，只是结点 5 和 6 在树中的位置发生了交换。
因为 5 和 6 的位置仍然相同，所以答案保持不变，仍然按值从小到大排序。
提示：
  • 树中结点数目总数在范围 [1, 1000] 内
  • 0 <= Node.val <= 1000
通过次数25,168提交次数47,193

BFS

思路
参考
🏒
651 · 二叉树垂直遍历 ，先将位置信息保存下来，然后排序，提取
题解
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def verticalTraversal(self, root: TreeNode) -> List[List[int]]:
        if root is None:
            return None
        # posiyion 存放（col, raw, val） 然后排序即可
        position = []
        queue = deque()
        queue.append((0, 0, root))
        while queue:
            col, raw, node = queue.popleft()
            position.append((col, raw, node.val))
            if node.left:
                queue.append((col-1, raw+1, node.left))
            if node.right:
                queue.append((col+1, raw+1, node.right))
        
        # 想办法将 position中的值取出来
        # 先排序，满足要求
        position.sort()
				# 找到最左边的位置，要从最左边开始
        min_pos = position[0][0]
        res = []
        temp = []
        for p in position:
            if p[0] == min_pos:
                temp.append(p[2])
            else:
                res.append(temp[:])
                temp = []
                temp.append(p[2])
                min_pos += 1
                
        res.append(temp[:])    
        return res