给定二叉树,按垂序遍历返回其结点值。
对位于 (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。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/vertical-order-traversal-of-a-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
定义一个节点:然后层序遍历(使用queue层序遍历;不明白的可以百度一下,很简单的)把节点的x,y,value存下来
排序:先按x排序,x相等按y排序,xy都相等按value排序
最后输出就行啦
/**
* 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:
struct node{
int xi;
int yi;
int value;
node(int X,int Y,int V):xi(X),yi(Y),value(V){}
};//该结点为树节点的x,y,value
struct tree{
int m;
int n;
TreeNode* p;
tree(int X,int Y,TreeNode* r):m(X),n(Y),p(r){}
};//该节点为层序遍历树节点时候使用的结点
static bool cmp(node a,node b){
if(a.xi==b.xi&&a.yi!=b.yi) return a.yi>b.yi;
else if(a.xi==b.xi&&a.yi==b.yi) return a.value<b.value;
else return a.xi<b.xi;
}
vector<vector<int>> verticalTraversal(TreeNode* root) {
// 先把每个结点的坐标存下来,然后再遍历
vector<vector<int>> ans;
queue<tree> Q;
vector<node> v;
Q.push(tree(0,0,root));
v.push_back(node(0,0,root->val));
tree k(0,0,root);
TreeNode *q;
int x,y;
while(!Q.empty()){
int len = Q.size();
for(int i=0;i<len;i++){
k = Q.front();
Q.pop();
x=k.m,y=k.n,q=k.p;
if(q->left){Q.push(tree(x-1,y-1,q->left));v.push_back(node(x-1,y-1,q->left->val));}
if(q->right){Q.push(tree(x+1,y-1,q->right));v.push_back(node(x+1,y-1,q->right->val));}
}
}
sort(v.begin(),v.end(),cmp);
vector<int> temp;
for(int i=0;i<v.size();i++){
// cout<<v[i].xxx<<" "<<v[i].yyy<<" "<<v[i].value<<endl;
temp.push_back(v[i].value);
if(i==v.size()-1||v[i].xi!=v[i+1].xi){
ans.push_back(temp);
temp.clear();
}
}
return ans;
}
};