记录关于二叉树中的广度优先搜索(遍历)方法及相关算法题目

   关于二叉树按广度优先搜索需要使用一个辅助队列数据结构来存储树的每层节点信息,主要原因是队列具有first in first out 性质,及符合一层层遍历树的逻辑,而栈结构具有first in last out即符合深度优先搜索策略。

按广度优先搜索策略遍历二叉树代码:

class Solution{
public:
	vector<vector<int> > levelOrder(TreeNode* root){
		queue<TreeNode*> que;
		if(root != nullptr) que.push(root);
		vector<vector<int>> result;
		while(!que.empty()){
			int size = que.size();
			vector<int> re;
			for (int i = 0;i< size;i++){
				TreeNode* node = que.front();
				que.pop;
				re.push_back(node->val);
				if (node->left != nullptr) que.push(node->left);
				if (node->right != nullptr) que.push(node->right);
				}
			result.push_back(re);
		}
		return result;
	}
}

变种:二叉树的层次遍历II, 给定一个二叉树,返回其节点值自底向上的层次遍历,(即按照从叶子节点所在层到根节点所在的层,逐层从左到右进行遍历)

在这里插入图片描述

class Solution{
public:
	 vector<vector<int> > levelOrderBottom(TreeNode* root){
		queue<TreeNode*> que;
		if (!root) que.push(root);
		vector<vector<int> > result;
		while(!que.empty()){
			int size = que.size();
			vector<int> vec;
			for (int i = 0;i< size;i++){
				TreeNode* node = que.front();
				que.pop();
				vec.push_back(node->val);
				if (node->left) que.push(node->left);
				if (node->right) que.push(ndoe->right);
			}
			result.push_back(vec);
		}
		reverse(result.begin(), result.end());//将得到的节点序列进行反转即可
		return result;
	}
}

变种:二叉书的右视图, 给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点的值。
在这里插入图片描述
思路:所有右边能看到的元素就是在每一层的最后的元素,在遍历每层元素的时候,判断当前元素是否为该层的最后的元素,若是的话,直接添加到结果vector中即可。
代码:

class Solution{
public:
	vector<int> rightSideView(TreeNode* root){
		queue<TreeNode*> que;
		if (root!= nullptr) que.push(root);
		vector<int> result;
		while(!que.empty()){
			int size = que.size();
			for (int i = 0;i< size;i++){
				TreeNode* node = que.front();
				que.pop();
				if (i == (size-1)) result.push_back(node->val);
				if (node->left) que.push(node->left);
				if (node->right) que.push(node->right); 
			}
		}
		return result;
	}
}

变种:二叉树层的平均值, 给定一个非空的二叉树,返回一个由每层节点的平均值组成的数组。
在这里插入图片描述
思路:逐层遍历,然后求取平均值即可。

class Solution{
public:
	vector<int> rightSideView(TreeNode* root){
		queue<TreeNode*> que;
		if (root!= nullptr) que.push(root);
		vector<int> result;
		while(!que.empty()){
			int size = que.size();
			int sum = 0
			for (int i = 0;i< size;i++){
				TreeNode* node = que.front();
				que.pop();
				sum += node->val;
				if (i == (size-1)) result.push_back(sum/size);
				if (node->left) que.push(node->left);
				if (node->right) que.push(node->right); 
			}
		}
		return result;
	}
}

变种:N叉树的层序遍历,给定一个N叉树,返回其节点值得层序遍历。
如下图所示给出一个N叉树,返回结果为:
在这里插入图片描述
返回结果:
[
[1],
[3,2,4],
[5,6]
]

代码:

class Solution {
public:
    vector<vector<int>> levelOrder(Node* root) {
        queue<Node*> que;
        if (root != NULL) que.push(root);
        vector<vector<int>> result;
        while (!que.empty()) {
            int size = que.size();
            vector<int> vec;
            for (int i = 0; i < size; i++) { 
                Node* node = que.front();
                que.pop();
                vec.push_back(node->val);
                for (int i = 0; i < node->children.size(); i++) { // 将节点孩子加入队列
                    if (node->children[i]) que.push(node->children[i]);
                }
            }
            result.push_back(vec);
        }
        return result;

    }
};

变种:在每个树的行中找最大值。
在这里插入图片描述

class Solution {
public:
    vector<int> largestValues(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        vector<int> result;
        while (!que.empty()) {
            int size = que.size();
            int maxValue = INT_MIN; // 取每一层的最大值
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                maxValue = node->val > maxValue ? node->val : maxValue;
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            result.push_back(maxValue); // 把最大值放进数组
        }
        return result;
    }
};

变种:填充每个节点的下一个右侧节点指针,当给定一个完美二叉树,其所有叶子节点都在同一层,每个父节点都有两个子节点。
此时二叉树的数据结构为:
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。初始状态下,所有 next 指针都被设置为 NULL。
示例:
在这里插入图片描述
思路:在遍历每一层节点的时候,记录本层的头部节点,然后依次让前一节点的next指针指向它的右侧节点。
代码:

class Solution {
public:
    Node* connect(Node* root) {
        queue<Node*> que;
        if (root != NULL) que.push(root);
        while(!queue.empty()){
        	int size = queue.size();
        	vector<int> vec;
        	Node* nodepre;
        	Node* node;
        	forint i = 0;i< size;i++{
				if (i==0){
					nodepre = que.front();//本层第一个节点
					que.pop();
					node = nodepre;
				}else{
					node = que.front();
					que.pop();
					nodepre->next = node;//前一个节点指向右边的节点
					nodepre = nodepre->next;
				}
				if (node->left) que.push(node->left);
				if (node->right) que.push(node->right);
			}
			nodepre->next = nullptr;//本层最后一个节点
        }
        return root;
};

变种:填充每个节点的下一个右侧节点指针II,跟上道题一样。

class Solution {
public:
    Node* connect(Node* root) {
        queue<Node*> que;
        if (root != NULL) que.push(root);
        while (!que.empty()) {
            int size = que.size();
            vector<int> vec;
            Node* nodePre;
            Node* node;
            for (int i = 0; i < size; i++) {
                if (i == 0) {
                    nodePre = que.front(); // 取出一层的头结点
                    que.pop();
                    node = nodePre;
                } else {
                    node = que.front();
                    que.pop();
                    nodePre->next = node; // 本层前一个节点next指向本节点
                    nodePre = nodePre->next;
                }
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            nodePre->next = NULL; // 本层最后一个节点指向NULL
        }
        return root;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值