题解 | #二叉树中和为某一值的路径(二)#

/**
 * struct TreeNode {
 *    int val;
 *    struct TreeNode *left;
 *    struct TreeNode *right;
 * };
 *
 * C语言声明定义全局变量请加上static，防止重复定义
 */
int g_node[1500] = {0};
int g_node_num = 0;
int get_tree_depth(struct TreeNode* root)
{
    if (root == NULL) {
        return 0;
    }
    int left_depth = get_tree_depth(root->left);
    int right_depth = get_tree_depth(root->right);
    return ((left_depth > right_depth) ? left_depth : right_depth) + 1;
}
void save_tree_node(struct TreeNode *root, int level)
{
    if (root == NULL) {
        return;
    }
    if (level == 0) {
        g_node[g_node_num++] = root->val;
    } else {
        save_tree_node(root->left, level - 1);
        save_tree_node(root->right, level - 1);
    }
}
/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param root TreeNode类 
 * @return int整型二维数组
 * @return int* returnSize 返回数组行数
 * @return int** returnColumnSizes 返回数组列数
 */
int** levelOrder(struct TreeNode* root, int* returnSize, int** returnColumnSizes ) {
    // write code here
    if (root == NULL) {
        return NULL;
    }
    (*returnSize) = 0;
    int** array = (int**)malloc(sizeof(int*) * 1500);  /* 最大1500结点 */
    (*returnColumnSizes) = (int*)malloc(sizeof(int) * 1500);

    int tree_depth = get_tree_depth(root);
    for (int i = 0; i < tree_depth; i++) {
        g_node_num = 0;
        save_tree_node(root, i);
        array[(*returnSize)] = (int*)malloc(sizeof(int) * g_node_num);
        memcpy(array[(*returnSize)], g_node, sizeof(int) * g_node_num);
        (*returnSize)++;
        (*returnColumnSizes)[i] = g_node_num;
    }
    return array;
}