[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<queue>
 
class Solution {
public:
    vector<int> largestValues(TreeNode* root) {
        queue<TreeNode*> Q;
        Q.push(root);
        int count_this = 1;
        int count_next = 0;
        vector<int> 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;
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值