leetcode 987. 二叉树的垂序遍历 - DFS -层次遍历 -测试用例 - 数组转二叉树

给你二叉树的根结点 root ，请你设计算法计算二叉树的 垂序遍历 序列。

对位于 (row, col) 的每个结点而言，其左右子结点分别位于 (row + 1, col - 1) 和 (row + 1, col + 1) 。树的根结点位于 (0, 0) 。

二叉树的 垂序遍历 从最左边的列开始直到最右边的列结束，按列索引每一列上的所有结点，形成一个按出现位置从上到下排序的有序列表。如果同行同列上有多个结点，则按结点的值从小到大进行排序。

返回二叉树的 垂序遍历 序列。

输入：root = [3,9,20,null,null,15,7]
输出：[[9],[3,15],[20],[7]]
解释：
列 -1 ：只有结点 9 在此列中。
列 0 ：只有结点 3 和 15 在此列中，按从上到下顺序。
列 1 ：只有结点 20 在此列中。
列 2 ：只有结点 7 在此列中。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/vertical-order-traversal-of-a-binary-tree

题目分析

找到根节点和左右孩子的坐标关系，然后可以用深度优先遍历和层次遍历给对应坐标赋值

DFS

class Solution:
    def verticalTraversal(self, root: TreeNode) -> List[List[int]]:
        mydic = {}
        a = b = 0
        def dfs(root,a,b):
            if not root: return
            if b not in mydic:mydic[b] = {}
            if a not in mydic[b]:mydic[b][a] = []
            mydic[b][a].append(root.val)
            dfs(root.left,a+1,b-1)
            dfs(root.right,a+1,b+1)
        dfs(root,a,b)
        ans = []
        for b in sorted(mydic):
            tp = []
            for a in sorted(mydic[b]):
                for x in sorted(mydic[b][a]):
                    tp.append(x)
            ans.append(tp)
        return ans

层次遍历

class Solution:
    def verticalTraversal(self, root: TreeNode) -> List[List[int]]:
        mydic = {}
        a = b = 0
        ans = []
        def layer(root):
            q = [(a,b,root)]
            while q:
                tp = []
                for a,b,x in q:
                    if b not in mydic:mydic[b]={}
                    if a not in mydic[b]:mydic[b][a] = []
                    mydic[b][a].append(x.val)
                    if x.left:tp.append((a+1,b-1,x.left))
                    if x.left:tp.append((a+1,b+1,x.right))
                q = tp[:]

测试用例 - 数组转二叉树

class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
def bfs(btree,i):
    if i >= len(btree):return None
    if btree[i] == None:return None
    root = TreeNode(btree[i])
    root.left = bfs(btree, 2*i+1)
    root.right = bfs(btree, 2 * i + 2)
    return root

btree = [3,9,20,None,None,15,7]
btree = [0,None,1]
root = bfs(btree,0)