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]
这道题的题意还是很简单的,给定一颗二叉树,求出每一层的节点最大值,将每层的最大值都存放在同一个向量中,最后返回向量。
可以用广搜和深搜两种方法解决此题,我选择了广搜解决此题,代码如下:
/**
* 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> max;
if(root == NULL) return max;
//vector<int> max;
queue<TreeNode*> q;
q.push(root);
int count_this = 1;
int count_next = 0;
//memset(count, 0, sizeof(count));
//int j = 0;
while(!q.empty()) {
int maxvalue = INT_MIN;
for(int i = 0; i < count_this; i++) {
TreeNode *temp = q.front();
q.pop();
if(temp->left != NULL) {
count_next++;
q.push(temp->left);
}
if(temp->right != NULL) {
count_next++;
q.push(temp->right);
}
if(temp->val > maxvalue) maxvalue = temp->val;
}
max.push_back(maxvalue);
count_this = count_next;
count_next = 0;
}
return max;
}
};
使用队列来进行广搜,count_this和count_next是用来记录当前层数和下一层的节点数目。比如说遍历到第二层,有两个节点,这两个节点都有左右子树,那么第三层就有四个节点,count_this作为当前for循环的比较值为2,count_next作为下一次for循环的比较值为4。最开始count_this的值为1,即根节点对应的层只有一个根节点。
还要注意的是本题测试样例比较极端,所以最开始maxvalue的初始值一定要设为INT_MIN,如果设为其他值比如说-1000,会遇到某一层的值都比-1000小(有的时候都接近INT_MIN),这时候传给向量的值就不是这层的最大值了,而是初始值-1000,就会有错误。