深度优先搜索系列八 LeetCode 515 在每个树行中找最大

2020-02-11  本文已影响0人  徐慵仙

题目

https://leetcode-cn.com/problems/find-largest-value-in-each-tree-row/

找每行的最大值

代码

/**
 * 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;
        help(root,0,res);
        return res;
    }
    void help(TreeNode* root,int k,vector<int>& res){
        if(root==NULL) return;
        if(k==res.size()) res.push_back(root->val);
        else res[k]=max(res[k],root->val);
        help(root->left,k+1,res);
        help(root->right,k+1,res);
    }
};

简析

基础题,此处用深度优先搜索解决,用一个向量res保存每一层的最大值,到达某个k层的某个节点时,如果res中还没有值,res[k]等于该节点的value,否则返回较大的一个。

上一篇下一篇

猜你喜欢

热点阅读