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;
}
};