Print A Tree In Vertical Order

最新推荐文章于 2023-07-20 01:16:17 发布

feliciafay

最新推荐文章于 2023-07-20 01:16:17 发布

阅读量676

点赞数

分类专栏： C++ InterviewQuestions 文章标签： Tree

本文链接：https://blog.csdn.net/feliciafay/article/details/44620049

版权

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

262 篇文章 1 订阅

订阅专栏

InterviewQuestions

183 篇文章 0 订阅

订阅专栏

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