515. 在每个树行中找最大值
解题思路
利用深度有限搜索算法。
vector类的size()函数返回该容器对象当前存储的元素个数。
当结果数组的size和二叉树的层数相同时,替换数据,使结果数组内数据值最大。
代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<int>ans;
//深度优先搜索
void dfs(TreeNode*p,int h){ //根节点 h=1
if(p==NULL)return;
//vector类的size()函数返回该容器对象当前存储的元素个数。
if(ans.size()<h)ans.push_back(p->val); //已放根节点值size=1,
else ans[h-1]=max(ans[h-1],p->val);
dfs(p->left,h+1); //查找左子树
dfs(p->right,h+1); //查找右子树
}
vector<int> largestValues(TreeNode* root) {
if(root==NULL)return ans;
ans.push_back(root->val); //根节点直接存入结果
dfs(root,1);
return ans;
}
};