Print A Tree In Vertical Order

最新推荐文章于 2021-08-03 09:14:59 发布

原创最新推荐文章于 2021-08-03 09:14:59 发布 · 702 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Tree

C++ 同时被 2 个专栏收录

262 篇文章

订阅专栏

InterviewQuestions

183 篇文章

订阅专栏

本文介绍了一种二叉树的垂直遍历算法，并通过示例详细解释了该算法的工作原理。文章提供了完整的C++代码实现，展示了如何确定每个节点与根节点的相对位置，进而按距离进行遍历。

参见 http://www.geeksforgeeks.org/print-binary-tree-vertical-order/

1 首先定义 Vertical Order,

1
/ \
2 3
/ \ / \
4 5 6 7
\ \
8 9

The output of print this tree vertically will be:
4 :
2
1 5 6
3 8
7
9

因为

4 : 与root的距离为-2
2: 与root的距离为-1
1 5 6: 与root的距离为0
3 8: 与root的距离为1
7: 与root的距离为2
9 : 与root的距离为3

所以按照和root的距离从远到近的顺序排列下来，就是上面所示的打印结果。

2 代码如下：

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
public:
    
    void get_min_max(TreeNode* root, int & min, int & max, int current_distance ) {
        if (root == NULL) return;
        if (current_distance < min) min = current_distance;
        if (current_distance > max) max = current_distance;
        get_min_max(root->left, min, max, current_distance - 1);
        get_min_max(root->right, min, max, current_distance + 1);
    }
    
    void helper(TreeNode* root, int line_index, int current_distance) {
        if (root == NULL) return;
        if(line_index == current_distance) {
            std::cout<<root->val<<", ";
        }
        helper(root->left, line_index, current_distance - 1);
        helper(root->right, line_index, current_distance + 1);
    }
    void vertical_tra(TreeNode* root) {
        int min;
        int max;
        int current_distance = 0;
        get_min_max(root, min, max, current_distance);
        std::cout<<"min = "<<min<<", max = "<<max<<std::endl;
        for (int i = min; i < max; ++i ) {
            helper(root, i, current_distance);
            std::cout<<std::endl;
        }
        
    }
};