987. 二叉树的垂序遍历/C++

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

给定二叉树，按垂序遍历返回其结点值。

对位于 (X, Y) 的每个结点而言，其左右子结点分别位于 (X-1, Y-1) 和 (X+1, Y-1)。

把一条垂线从 X = -infinity 移动到 X = +infinity ，每当该垂线与结点接触时，我们按从上到下的顺序报告结点的值（
Y 坐标递减）。

如果两个结点位置相同，则首先报告的结点值较小。

按 X 坐标顺序返回非空报告的列表。每个报告都有一个结点值列表。

示例 1：

输入：[3,9,20,null,null,15,7] 输出：[[9],[3,15],[20],[7]] 解释：
在不丧失其普遍性的情况下，我们可以假设根结点位于 (0, 0)： 然后，值为 9 的结点出现在 (-1, -1)； 值为 3 和 15
的两个结点分别出现在 (0, 0) 和 (0, -2)； 值为 20 的结点出现在 (1, -1)； 值为 7 的结点出现在 (2,
-2)。
示例 2：

输入：[1,2,3,4,5,6,7] 输出：[[4],[2],[1,5,6],[3],[7]] 解释： 根据给定的方案，值为 5 和 6
的两个结点出现在同一位置。 然而，在报告 “[1,5,6]” 中，结点值 5 排在前面，因为 5 小于 6。

提示：

树的结点数介于 1 和 1000 之间。 每个结点值介于 0 和 1000 之间。

代码参考了zhou_zzz的评论
层次遍历的顺序就可以代表纵坐标，因此我们的结构体就只需要增添横坐标即可。剩下的看代码注释即可。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> verticalTraversal(TreeNode* root) {
        vector<vector<int>> res;
        if(root==NULL)
            return res;
        
        
        map<int,vector<int>> mapiv;		//这就是我们用的结构体，int代表横坐标，vector包含一列所包含的节点的值
        queue<pair<int,TreeNode*>> que;		//层次遍历所需的队列，这两个结构体都增加了一个int，代表横坐标
        que.push(make_pair(0,root));
        while(!que.empty()){ 
            map<int,vector<int>> tmp;
            int size = que.size();
            for(int i=0;i<size;i++){
                pair<int,TreeNode*> cur = que.front();
                que.pop();
                tmp[cur.first].push_back(cur.second->val);	//向代表横坐标为cur.first的vector里加入节点的值
                
                if(cur.second->left)
                    que.push(make_pair(cur.first-1,cur.second->left));	//左边的横坐标减一
                if(cur.second->right)
                    que.push(make_pair(cur.first+1,cur.second->right));	//右边的横坐标加一
            }
            //由于是在同一层，也就是y相等，那么每一个a中的vector的元素都是重合的节点，因此需要排序。
            //之后就可以加入到mapiv中
            for(auto a:tmp){   
                sort(a.second.begin(),a.second.end());
                mapiv[a.first].insert(mapiv[a.first].end(),a.second.begin(),a.second.end());
            }
        }
        for(auto a:mapiv){
            res.push_back(a.second);
        }
        return res;
    }
};