深度优先搜索系列八 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,否则返回较大的一个。