给你一棵二叉树的根节点 root ,请你构造一个下标从 0 开始、大小为 m x n 的字符串矩阵 res ,用以表示树的 格式化布局 。构造此格式化布局矩阵需要遵循以下规则:
树的 高度 为 height ,矩阵的行数 m 应该等于 height + 1 。
矩阵的列数 n 应该等于 2height+1 - 1 。
根节点 需要放置在 顶行 的 正中间 ,对应位置为 res[0][(n-1)/2] 。
对于放置在矩阵中的每个节点,设对应位置为 res[r][c] ,将其左子节点放置在 res[r+1][c-2height-r-1] ,右子节点放置在 res[r+1][c+2height-r-1] 。
继续这一过程,直到树中的所有节点都妥善放置。
任意空单元格都应该包含空字符串 “” 。
返回构造得到的矩阵 res 。
来源:力扣(LeetCode)
解答:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int longHeight(TreeNode* root){
int l = 0, r = 0;
if(root->left!= nullptr){
l = max(l, longHeight(root->left)+1);
}
if(root->right!=nullptr){
r = max(r, longHeight(root->right)+1);
}
return max(l, r);
}
void dfs(vector<vector<string>>& ans, TreeNode* node, int r, int c, const int& height){
ans[r][c] = to_string(node->val);
if(node->left){
dfs(ans, node->left, r+1, c - (1 << (height-r-1)), height);
}
if(node->right){
dfs(ans, node->right, r+1, c +(1 << (height-r-1)), height);
}
}
vector<vector<string>> printTree(TreeNode* root) {
int h = longHeight(root);
int m = h + 1;
int n = (1 << (h + 1)) - 1;
vector<vector<string>> ans(m, vector<string>(n, ""));
dfs(ans, root, 0, (n - 1)/2, h);
return ans;
}
};