第十二周算法分析与设计： Find Largest Value in Each Tree Row

问题描述：

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

Example:

Input:

1
/ \
3   2
/ \   \
5   3   9

Output: [1, 3, 9]

解法思路：

跟第十周的算法思路类似，在层次遍历时记下每个节点——当到达新一层时，将当前节点的值放入vector中；在之后的同层遍历中，如果有比它大的就替换。

class Solution {
public:
vector<int> largestValues(TreeNode* root) {
vector<int> result;
if(root != NULL) {
queue<int>level;
queue<TreeNode*>node;
node.push(root);
level.push(1);
result.push_back(root->val);
int m = 1;
while(!node.empty()) {
TreeNode* cur_n = node.front();
int cur_l = level.front();
if(cur_n->left) {
node.push(cur_n->left);
level.push(cur_l+1);
}
if(cur_n->right) {
node.push(cur_n->right);
level.push(cur_l+1);
}
node.pop();
level.pop();
if(m == cur_l) {
if(cur_n->val > result[m-1])
result[m-1] = cur_n->val;
}
else {
m++;
result.push_back(cur_n->val);
}
}
}
return result;
}
};

算法的执行时间复杂度为O(n)。

休息下，这波不亏~