515. Find Largest Value in Each Tree Row

题目

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

想法

嗯,有一个vector来记录每一层的最大值,然后搜一遍树,更新最大值就可以了,应该就是这样

我的答案

一次AC了

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
/**
* 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> maxVal;
vector<int> largestValues(TreeNode* root) {
travel(root, 1);
return maxVal;
}
void travel(TreeNode* node, int level) {
if (node == NULL)
return;
if (maxVal.size() < level)
maxVal.push_back(node->val);
else if (node->val > maxVal[level - 1])
maxVal[level - 1] = node->val;
travel(node->left, level + 1);
travel(node->right, level + 1);
}
};

回顾

比较明确,比较简单吧。
不做多余的事,更新最大值就好了,难得有一题如此轻松地AC哈哈。