二叉树的垂直遍历-编程知识

1.题目

这道题是2024-2-13的签到题，题目难度为困难。

考察的知识点是DFS算法和自定义排序。

题目链接：二叉树的垂直遍历

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

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

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

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

2.思路

说实话，之前碰到的困难题目大部分我都是直接放弃的，但是通过这段时间对DFS算法的使用后我尝试独立解决这道题，发现还是可以解出来的。这道题其实有两大难点：

如何得到所有列的所有结点
如何让每列的结点按照题目要求排序

下面是我对这道题的所有思路：

初步思路——如何理解坐标

对于题目中的要求，根节点的坐标为(0,0)，其它坐标都符合以下规律：

对于所有的左子结点，它的坐标为(x_parent+1,y_parent-1)。

对于所有的右子结点，它的坐标为(x_parent+1,y_parent+1)。

其中x_parent是这些节点的父节点x坐标值，y_parent是这些结点的父节点y坐标值。

选择什么算法来解题——DFS算法

其实刚开始我是打算用BFS算法来解题的，因为根据这道题的结果来看，它里面的列表结果值是根据结点的x坐标排序的，但如果使用BFS算法有一个问题，就是我如何在BFS遍历过程中读取父节点的坐标呢？因为这个坐标是我们自己定义的，并不是TreeNode自带的属性。因此我想到了使用DFS遍历。

对于DFS遍历，它一般都是用递归的方式来写的，因此我可以给递归函数加上两个参数用来传入父节点的x坐标和y坐标。同时我们根据前面的坐标规律来进行子结点遍历。对于每个结点，我们利用一个结构来存储它们的信息——字典（哈希表）。这样我们就可以把每个列的所有结点信息保留。

怎么按照题目要求排序？

在完成DFS遍历所有节点后我们能够得到所有列的所有结点，但这就完事了吗？它还有一个难点，我们还需要进行排序，因此我们得弄清楚题目的排序要求：

根据列的值来排序（升序）
根据行的值来排序（升序）
根据结点的值排序（升序）

由于我比较喜欢用python来写算法题目，因此这里我借助python的sorted函数、lambda表达式等python内置语法来写一些排序规则，下面是我的排序整体思路：

首先我们用字典的内置函数items()函数来获取一个迭代器对象，并将这个迭代器传入到sorted函数里面（进行排序），排序规则按照字典的键来升序排序。

将前面得到的迭代器对象根据结点的列排序后，我们还需要根据后面两个排序规则排序，这里我们利用python的for循环来获取前面的迭代器对象元素，同时我们还要借助元组的拆包语法来获取对应的两个子元素：sk,val。

我们需要对val进行自定义排序，排序规则根据每个元素的行和值进行排序。最后我们利用列表表达式来得到每一列的所有节点值，并存入到结果列表中。

最后我们直接返回结果列表就行了。

3.代码

# 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: Optional[TreeNode]) -> List[List[int]]:# 定义一个字典，用来存储每一列的结点信息rst = {}# 定义DFS函数，这里是前序遍历def dfs(node,x,y):# 如果结点存在的话if node:# 如果rst里面没有第y列的话，则初始化第y列的列表信息# 其中，x是当前结点的行，node.val是当前结点的值if y not in rst:rst[y] = [(x,node.val)]# 如果rst里面有第y列的话，则添加第y列的列表信息else:rst[y].append((x,node.val))# 剪枝操作，如果当前结点存在左子结点if node.left:dfs(node.left,x+1,y-1)# 剪枝操作，如果当前结点存在右子结点if node.right:dfs(node.right,x+1,y+1)# DFS遍历root结点dfs(root,0,0)# 获取前面的字典迭代器对象（排序后的）# 根据列来排序sort_key = sorted(rst.items(),key=lambda x:x[0])# 保存最终结果ans = []# 利用拆包和循环遍历前面的迭代器对象for sk,val in sort_key:# 将每列的结点信息按照行和它的值来排序val = sorted(val,key=lambda x:[x[0],x[1]])# 利用列表推导式来获取最终的格式结果tmp = [v[1] for v in val]# 添加到结果列表里面ans.append(tmp)return ans

4.总结

今天的这道签到题是今年（2024）我第一道有成就感的题目，毕竟是今年第一道自己独立解决的困难题。过年的这几天签到题都是DFS算法或BFS算法，这些天的练习让我对这两种算法有更深的理解。今天的这道题更是让我对前些天的练习有一个综合运用。也让我对之后的算法练习和学习有更大的鼓舞，算法这个东西如果我们能够凭借自己的努力解出来那是相当有成就感的，上次有这种成就感还是大一的时候做高数题。

总而言之，算法思维并不是天生就有的，我也并不是天赋很好的选手，只能凭借每天的算法练习来学习和掌握常见的算法题。俗话说一个人走的更快，一群人走的更远。这段时间的算法练习中，我将自己每天的算法思路分享给大家参考，这不仅是让我对这道题进一步的分析，也能够给大家一点参考，最后祝大家新年快乐！。