leetcode 655 输出二叉树

leetcode 655 输出二叉树

在一个 m*n 的二维字符串数组中输出二叉树，并遵守以下规则：

• 行数 m 应当等于给定二叉树的高度。
• 列数 n 应当总是奇数。
• 根节点的值（以字符串格式给出）应当放在可放置的第一行正中间。根节点所在的行与列会将剩余空间划分为两部分（左下部分和右下部分）。你应该将左子树输出在左下部分，右子树输出在右下部分。左下和右下部分应当有相同的大小。即使一个子树为空而另一个非空，你不需要为空的子树输出任何东西，但仍需要为另一个子树留出足够的空间。然而，如果两个子树都为空则不需要为它们留出任何空间。
• 每个未使用的空间应包含一个空的字符串""。
• 使用相同的规则输出子树。

分析

• m 是树的深度，使用depth获取,注意从1开始
• n 是类似宽度的概念，$n=2^m-1$
• 通过层次遍历可以直接获取的节点信息有：level
• 还需要pos 才能确定节点的位置，那么pos怎么来的呢？
• 注意到 第一层中的节点位置为 $n1=(0+n)/2$, 第二层中left节点位置：$n2l=(0+n1)/2$,right节点位置：$n2r=(n1+n)/2$
• 所以每个节点需要额外携带一些信息：从父节点继承的scope信息，以确定位置。

code

class Solution {
    struct Node{
        int level;
        int l;  // left of scope
        int r;  // right of scope
        TreeNode* node;
        Node(int le,int l1,int r1,TreeNode* n):level(le),l(l1),r(r1),node(n){}
    };
    int depth(TreeNode* node){
        if(!node) return 0;
        return max(depth(node->left),depth(node->right))  +1;
    }
public:
    vector<vector<string>> printTree(TreeNode* root) {
        if(!root) return {{""}};
        int m = depth(root);
        int n = pow(2,m) -1;
        vector<vector<string>> result(m,vector<string>(n));
        queue<Node> level_queue;
        level_queue.push(Node(0,0,n,root));
        while(!level_queue.empty()){
            auto t = level_queue.front();
            level_queue.pop();
            int pos = (t.l + t.r) / 2;
            result[t.level][pos] = to_string(t.node->val);
            // 继承scope 信息
            if(t.node->left){
                level_queue.push(Node(t.level+1,t.l,pos,t.node->left));
            }
	        if(t.node->right){
	            level_queue.push(Node(t.level+1,pos,t.r,t.node->right));
	        }

        }
        return result;
    }
};