655. 输出二叉树

luckycoding

于 2021-03-31 00:03:22 发布

阅读量92

点赞数

文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/luckycoding/article/details/115339434

版权

这篇博客介绍了如何使用C++实现二叉树的层次遍历，并找到树的最大深度。通过递归函数计算每个节点的层级，然后用宽度优先搜索（BFS）填充二维字符串数组来展示树的结构。同时，代码中展示了如何处理空节点的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路先找深度，在递归赋值。

/**
 * 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:
    int get_tree_level(TreeNode *root, int level)
    {
        if (root == NULL)
            return level;
        return max(get_tree_level(root->left, level + 1), get_tree_level(root->right, level + 1));
    }
    void solve(TreeNode *root, vector<vector<string>> &ans, int x, int y, int m)
    {
        if (root == NULL)
            return  ;
        ans[x][y] = to_string(root->val);
        m = m - (m >> 1);
        cout << m << endl;
        solve(root->left, ans, x + 1, y - m, m);
        solve(root->right, ans, x + 1, y + m, m);
    }
    vector<vector<string>> printTree(TreeNode* root) {
        vector<vector<string>> ans;
        int level = get_tree_level(root, 0);
        // cout << level << endl;
        const int m = (1 << level) - 1; 
        for (int i = 0; i < level; i++)
            ans.push_back(vector<string>(m, ""));
        solve(root, ans, 0, (1 << (level - 1)) - 1,  (1 << (level - 1)) - 1);
        return ans;
    }
};
/*
1 1 0 
2 3 1
3 7 3
4 15 7
*/