【Leetcode】314. Binary Tree Vertical Order Traversal-CSDN博客

本文链接：https://blog.csdn.net/qq_46105170/article/details/105862006

本文介绍了解决LeetCode上二叉树垂直遍历问题的两种方法：一种使用DFS结合map实现，另一种使用BFS结合map实现。通过这两种方法，可以有效地按列收集二叉树节点，并保持节点的上下左右相对位置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址：

https://leetcode.com/problems/binary-tree-vertical-order-traversal/

给定一个二叉树，将其每一列的节点按照从上到下从左到右的顺序存成一个列表返回。

BFS。BFS可以有效解决两个节点谁在上谁在下，和谁在左谁在右的问题，但没法判断两个节点是否在同一列。可以新开一个类，里面存树的节点和它在的列，然后用一个map以列为key，该列的节点的val为值，最后将这个map遍历一遍加入最终结果即可。代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left),
 * right(right) {}
 * };
 */
class Solution {
 public:
  using PTI = pair<TreeNode*, int>;
  vector<vector<int>> verticalOrder(TreeNode* root) {
    vector<vector<int>> res;
    if (!root) return res;

    unordered_map<int, vector<int>> mp;
    queue<PTI> q;
    q.push({root, 0});
    int min_col = 0;
    while (q.size()) {
      auto p = q.front(); q.pop();
      min_col = min(min_col, p.second);
      auto* t = p.first;
      mp[p.second].push_back(t->val);
      if (t->left) q.push({t->left, p.second - 1});
      if (t->right) q.push({t->right, p.second + 1});
    }
    res.resize(mp.size());
    for (auto& [k, v] : mp) res[k - min_col].swap(v);
    return res;
  }
};

时空复杂度 $O (n)$ 。