[DFS]515. Find Largest Value in Each Tree Row

题目：

题目分析：

1、题目要求通过给出的二叉树找出每行最大值。最后通过一个容器返回所有结果。

2、这里注意找的是每一行的最大值，我们知道对于一个数组而言，最大值仅需遍历一遍即可；那么我们把每棵树当作若干数组的堆叠，仅需对每层进行一次遍历即可。很显然层序遍历的方法就是BFS。

3、明确了通过BFS求解后，还需要解决细节上的问题。比如，在BFS的队列实现中，如何找到各层的起始和终止位置。本题中我采用的方法就是设置两个计数量count_this、count_next分别保存对本层元素个数和下层元素个数。初始本层个数为1（树的根节点），下层个数置零。那么在BFS过程中每层都通过count_this的循环来遍历，每在队尾放入一个元素count_next++，并在每层统计出最大元素max放入答案容器中。每层遍历结束时，将count_next的值交给count_this，count_next再置零以完成层遍历的更迭。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
#include
 
class Solution {
public:
    vector largestValues(TreeNode* root) {
        queue Q;
        Q.push(root);
        int count_this = 1;
        int count_next = 0;
        vector res;
        while(!Q.empty()){
        	int max = 0;
        	for(int i = 0; i < count_this; i++){
        		TreeNode* node = Q.front();
        		if(node->val > max)
        			max = node->val;
				if(node->left == NULL && node->right == NULL)
					continue;
				else{
					if(node->left != NULL && node->right == NULL){
						Q.push(node->left);
						count_next++;
					}
					else if(node->left == NULL && node->right != NULL){
						Q.push(node->right);
						count_next++;
					}
					else{
						Q.push(node->left);
						Q.push(node->right);
						count_next+=2;
					}
        		}
        		Q.pop();
			}
			res.push_back(max);
			count_this = count_next;
			count_next = 0;
        }
        return res;
    }
};