leetcode 987. 二叉树的垂序遍历(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 之间。

解法:

#define PR pair<int, int>
#define PPR pair<PR, int>
/**
 * 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:
    
    void traversal(TreeNode* root, int x, int y, vector<PPR>& res){
        if(root != NULL){
            res.push_back({{x, y}, root->val});
            traversal(root->left, x-1, y+1, res);
            traversal(root->right, x+1, y+1, res);
        }
    }
    
    vector<vector<int>> verticalTraversal(TreeNode* root) {
        vector<PPR> lst;
        vector<vector<int>> res;
        int x = 0, y = 0;
        traversal(root, x, y, lst);
        sort(lst.begin(), lst.end());
        int pre = lst[0].first.first;
        vector<int> tmp;
        for(PPR pr : lst){
            if(pr.first.first == pre){
                tmp.push_back(pr.second);
            }else{
                res.push_back(tmp);
                tmp.clear();
                tmp.push_back(pr.second);
                pre = pr.first.first;
            }
        }
        res.push_back(tmp);
        return res;
    }
};

猜你喜欢

转载自www.cnblogs.com/zhanzq/p/10916115.html