515 Find Largest Value in Each Tree Row

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

int height(struct TreeNode* node) {
    if (node == NULL) return 0;
    int l = height(node->left);
    int r = height(node->right);
    return (l > r ? l + 1 : r + 1);
}

int min(struct TreeNode* node) {
    int res = node->val;
    if (node->left) {
        int l = min(node->left);
        res = res > l ? l : res;
    }
    if (node->right) {
        int r = min(node->right);
        res = res > r ? r : res;
    }
    return res;
}

void traverse(struct TreeNode* node, int layer, int* res) {
    if (node == NULL) return;
    if (res[layer] < node->val) {
        res[layer] = node->val;
    }
    traverse(node->left, layer + 1, res);
    traverse(node->right, layer + 1, res);
    return;
}

/**
 * Return an array of size *returnSize.
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* largestValues(struct TreeNode* root, int* returnSize) {
    *returnSize = height(root);
    int* res = (int*) malloc((*returnSize) * sizeof(int));
    if (0 == *returnSize) return res;
    int m = min(root);
    for (int i = 0; i < *returnSize; i++) {
        res[i] = m;
    }
    traverse(root, 0, res);
    return res;
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容