【Leetcode】987. Vertical Order Traversal of a Binary Tree

题目地址：

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

给定一棵二叉树，要求垂直对其进行遍历，想象树根的坐标是$(0,0)$，对于每个节点$(x,y)$，其左右孩子的坐标分别是$(x+1,y-1)$和$(x+1,y+1)$。遍历的结果应当是若干列表形成的一个大的列表，这些列表按$y$的次序是从小到大。每个列表内部的数字应该按其在树中的深度由小到大排序。如果两个节点的坐标相等，则应该数值小的在前大的在后。

可以参考https://blog.csdn.net/qq_46105170/article/details/108745088的思路。下面介绍一个先汇总再排序的思路。

先DFS，用一个map将所有节点所在的列作为key，value是节点的行和值，然后遍历map，map已经按照节点的列从小到大排了序，接着对每个entry的value排序，那么相同列的节点会先按照行排序，然后按值排序，最后将这个entry的value的节点的值存入答案。代码如下：

/**
 * 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 PII = pair<int, int>;
  vector<vector<int>> verticalTraversal(TreeNode* root) {
    map<int, vector<PII>> mp;
    dfs(root, mp, 0, 0);
    vector<vector<int>> res;
    for (auto& [k, v] : mp) {
      sort(v.begin(), v.end(), [&](auto& p1, auto& p2) {
        if (p1.first != p2.first) return p1.first < p2.first;
        return p1.second < p2.second;
      });
      res.push_back({});
      for (auto& w : v) res.back().push_back(w.second);
    }
    return res;
  }

  void dfs(TreeNode* p, map<int, vector<PII>>& mp, int x, int y) {
    if (!p) return;
    mp[y].push_back({x, p->val});
    dfs(p->left, mp, x + 1, y - 1);
    dfs(p->right, mp, x + 1, y + 1);
  }
};

时间复杂度$O(n\log n)$，空间$O(n)$。