515. Find Largest Value in Each Tree Row

无差别刷题

于 2019-06-08 07:00:00 发布

阅读量71

点赞数

CC 4.0 BY-SA版权

文章标签： tree

本文链接：https://blog.youkuaiyun.com/wilzxu/article/details/91268202

本文介绍了一种使用深度优先搜索(DFS)算法来找到二叉树每一层的最大值的方法。通过递归地遍历树的左子树和右子树，并跟踪每个深度的最大值，可以有效地解决这个问题。

515. Find Largest Value in Each Tree Row

方法1: dfs

You need to find the largest value in each row of a binary tree.

Example:
Input:

Output: [1, 3, 9]

方法1: dfs

思路：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> largestValues(TreeNode* root) {
        vector<int> res;
        dfs(root, 0, res);
        return res;
    }
    
    void dfs(TreeNode* root, int level, vector<int> & res) {
        if (!root) return;
        if (level == res.size()) res.push_back({INT_MIN});
        res[level] = max(res[level], root -> val);
        dfs(root -> left, level + 1, res);
        dfs(root -> right, level + 1, res);
        return;
    }
};