LeetCode_102. 二叉树的层次遍历

#题目
102. 二叉树的层次遍历

给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。

例如:
给定二叉树: [3,9,20,null,null,15,7],

3
/ \
9 20
/ \
15 7

返回其层次遍历结果:

[
[3],
[9,20],
[15,7]
]

#题解

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
/**
 * Return an array of arrays of size *returnSize.
 * The sizes of the arrays are returned as *columnSizes array.
 * Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
 */
int max(int a, int b) {
    return a > b ? a : b;
}

int get_depth(struct TreeNode *root) {
    if (root == NULL) return 0;
    return max(get_depth(root->left), get_depth(root->right)) + 1;
}

void get_size(struct TreeNode *root, int *columnSizes, int level) {
    if (root == NULL) return ;
    ++columnSizes[level];
    get_size(root->left, columnSizes, level + 1);
    get_size(root->right, columnSizes, level + 1);
}

void get_val(struct TreeNode *root, int *columnSizes, int **data, int level) {
    if (root == NULL) return ;
    data[level][columnSizes[level]++] = root->val;
    get_val(root->left, columnSizes, data, level + 1);
    get_val(root->right, columnSizes, data, level + 1);
}

int** levelOrder(struct TreeNode *root, int **columnSizes, int *returnSize) {
    *returnSize = get_depth(root);
    *columnSizes = (int *)calloc((*returnSize), sizeof(int));
    int **data = (int **)malloc(sizeof(int *) * (*returnSize)); 
    get_size(root, *columnSizes, 0);
    for (int i = 0; i < *returnSize; i++) {
        data[i] = (int *)malloc(sizeof(int ) * (*columnSizes)[i]);
        (*columnSizes)[i] = 0;
    }
    get_val(root, *columnSizes, data, 0);
    return data;
}

LeetCode练习汇总

你可能感兴趣的:(数据结构,c/c++,LeetCode)