leetcode：314. 二叉树的垂直遍历

题目来源

• leetcode：314-VIP . 二叉树的垂直遍历 binary-tree-vertical-order-traversal
• lintcode：651 · 二叉树垂直遍历

题目描述

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:
    vector<vector<int>> verticalOrder(TreeNode* root){
    
    }
};

题目分析

二叉树题目的递归解法可以分为两类，第一类是遍历一遍二叉树得出答案，第二类是通过分解问题计算出答案，这两类思路分别对应着回溯算法和动态规划。

我们用递归的思路来想一想。首先一定要先遍历一遍树，然而树的遍历方法有四种：前序、中序、后序、层序，那应该是哪一种呢？我们仔细观察一下

它不是前中后序，所以应该和层序有关。那怎么确定列的顺序呢？

• 这里可以把根节点给个序号0，然后开始层序遍历，凡是左子节点则序号减一，右子节点序号加一，这样可以通过序号来把相同列的节点值放在一起
• 我们可以用一个Map来建立序号和其对应的节点值映射，这样可以用其自动排序功能让列自动从左到右。
• 由于层序遍历要用到queue，此时的queue不能只存节点，必须存储pair{序号，节点}，这样取出就可以操作序号，而且排入队列中的节点也能附上正确的序号

class Solution {
public:
    vector<vector<int>> verticalOrder(TreeNode* root){
        if(root == NULL){
            return {};
        }
        
        std::map<int, std::vector<int>> m;
        std::queue<std::pair<int, TreeNode*>> q;
        q.push({0, root});
        while (!q.empty()){
            auto a = q.front(); q.pop();
            m[a.first].push_back(a.second->val);
            if(a.second->left){
                q.push({a.first - 1, a.second->left});
            }
            if(a.second->right){
                q.push({a.first + 1, a.second->right});
            }
        }
        std::vector<std::vector<int>> res;
        for(auto &a : m){
            res.push_back(a.second);
        }
        return res;
    }
};