leetcode_655. Print Binary Tree

转载于 2019-04-12 11:30:00 发布 · 92 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/jasonlixuetao/p/10694984.html

文章标签：

#数据结构与算法

本文提供了一个解决LeetCode上打印整棵二叉树问题的C++代码实现。通过计算二叉树的高度和宽度，使用深度优先搜索算法，将二叉树的节点值按层级打印出来，形成一个矩阵形式的输出。

https://leetcode.com/problems/print-binary-tree/

打印整棵二叉树

class Solution
{
public:

    int getTreeHeight(TreeNode* root)
    {
        return root==NULL?0:max(getTreeHeight(root->left), getTreeHeight(root->right))+1;
    }

    vector<vector<string>> printTree(TreeNode* root)
    {
        int height = getTreeHeight(root);
        vector<vector<string>> res(height, vector<string>((int)pow(2, height)-1,""));
        dfs(root, 0 , 0, pow(2, height)-1, res);
        return res;
    }

    void dfs(TreeNode* root, int layer, int l, int r, vector<vector<string>>& res)
    {
        if(root == NULL)
            return;
        int mid = (l+r)/2;
        stringstream ss;
        string value;
        ss<<root->val;
        ss>>value;
        res[layer][mid] = value;
        dfs(root->left, layer+1, l, mid-1, res);
        dfs(root->right, layer+1, mid+1, r, res);
    }
};