515. 在每个树行中找最大值

LIZHUOLONG1

于 2024-05-15 15:51:20 发布

阅读量314

点赞数 4

分类专栏： LeetCode刷题文章标签：算法

本文链接：https://blog.csdn.net/LIZHUOLONG1/article/details/138911625

版权

LeetCode刷题专栏收录该内容

304 篇文章 4 订阅

订阅专栏

515. 在每个树行中找最大值

#define MAX_NODE_SIZE 10001
#define MAX(a, b) ((a) > (b) ? (a) : (b))

void dfs(int *res, int *pos, struct TreeNode* root, int curHeight) {
    if (curHeight == *pos) {
        res[(*pos)++] = root->val;
    } else {
        res[curHeight] = MAX(res[curHeight], root->val);
    }
    if (root->left) {
        dfs(res, pos, root->left, curHeight + 1);
    }
    if (root->right) {
        dfs(res, pos, root->right, curHeight + 1);
    }
} 

int* largestValues(struct TreeNode* root, int* returnSize) {
    if (!root) {
        *returnSize = 0;
        return NULL;
    }
    int *res = (int *)malloc(sizeof(int) * MAX_NODE_SIZE);
    *returnSize = 0;
    dfs(res, returnSize, root, 0);
    return res;
}