leetcode---987. 二叉树的垂序遍历

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。

/**
 * 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 x,y,val;
		};
		vector<node> vn;
		static bool cmp(node & a1,node & a2) {
			if(a1.x!=a2.x)
				return a1.x<a2.x;
			else if(a1.y!=a2.y)
				return a1.y>a2.y;
			else return a1.val<a2.val;
		}
		vector<vector<int>> verticalTraversal(TreeNode* root) {

			dfs(0,0,root);
			vector<vector<int> > res;
			sort(vn.begin(),vn.end(),cmp);
			if(vn.size()>0) 
                       {
				int t=vn[0].x;
				vector<int> tmp;
				tmp.push_back(vn[0].val);
				for(int i=1; i<vn.size(); i++) {
					if(vn[i].x==t) {
						tmp.push_back(vn[i].val);
					} else {
						res.push_back(tmp);
						tmp.clear();
						t=vn[i].x;
						tmp.push_back(vn[i].val);
					}
				}
				res.push_back(tmp);

			}
			return res;

		}
		void    dfs(int x,int y,TreeNode* root) {
			if(root!=NULL) {
				node t;
				t.x=x;
				t.y=y;
				t.val=root->val;
				vn.push_back(t);
				if(root->left!=NULL) {
					dfs(x-1,y-1,root->left);
				}
				if(root->right!=NULL) {
					dfs(x+1,y-1,root->right);
				}
			}
		}
};