数据结构之BFS广度优先算法(腐烂的苹果)

队列这个数据结构在很多场景下都有使用,比如在实现二叉树的层序遍历,floodfill问题(等等未完成)中,都需要借助队列的先进先出特性,下面给出这几个问题的解法

经典的二叉树的层序遍历

算法图示,以下图所示的二叉树为例

二叉树层序遍历 C++代码实现

cpp 复制代码
typedef struct binaryTreeNode {
	struct binaryTreeNode* left;
	struct binaryTreeNode* right;
	BTDataType data;
}BTNode;

void levelOrder(BTNode* root)
{
	std::queue<BTNode*> nodeQ;
	if (root)
	{
		nodeQ.push(root);
	}
	while (!nodeQ.empty())
	{
		BTNode* front = nodeQ.front();
		std::cout << front->data;
		if (front->left)
		{
			nodeQ.push(front->left);
		}
		if (front->right)
		{
			nodeQ.push(front->right);
		}
		nodeQ.pop();
	}
}

注意:队列中存放的是二叉树中结点的指针,每次都去找队列的头部元素,通过头部元素找到其非空的左右孩子,并把头部元素出队列,此时出队列得到的序列就是二叉树的层序遍历序列

Floodfill问题之腐烂的苹果

(牛客网链接:腐烂的苹果_牛客题霸_牛客网 (nowcoder.com)

Flood Fill(泛洪填充)算法是一种图像处理的基本算法,用于填充连通区域。该算法通常从一个种子点开始,沿着种子点的相邻像素进行填充,直到遇到边界或者其他指定的条件为止。

如果用BFS的思想来解决这个问题,每次循环都只考虑队列中头元素的上下左右

需要考虑的有:

  1. 查看上下左右元素时不能发生越界

  2. 只将上下左右元素中为 1 的那些元素的二维坐标位置放到队列中

  3. 放到队列中的二维坐标位置设置为 true

代码实现

cpp 复制代码
int rotApple(vector<vector<int> >& grid) {
    int upanddown[4] = {0, 0, -1, 1};
    int leftandright[4] = {-1, 1, 0, 0};
    bool vis[1010][1010] = {false};
    int n = grid.size();
    int m = grid[0].size();
    queue<pair<int, int>> findtwoq;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (grid[i][j] == 2) {
                findtwoq.push({i, j});
            }
        }
    }

    int ret = 0;
    while (findtwoq.size()) {
        int sz = findtwoq.size();
        ret++;
        while (sz--) {
            auto [a, b] = findtwoq.front();
            findtwoq.pop();
            for (int i = 0; i < 4; i++) {
                int x = a + upanddown[i], y = b + leftandright[i];
                if (x >= 0 && x < n && y >= 0 && y < m && grid[x][y] == 1 && !vis[x][y]){
                    vis[x][y] = true;
                    findtwoq.push({x, y});
                }
            }
        }
    }
    
    for(int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (grid[i][j] == 1 && vis[i][j] == false) {
                return -1;
            }
        }
    }
    return ret - 1;
}
相关推荐
Steve lu11 分钟前
回归任务损失函数对比曲线
人工智能·pytorch·深度学习·神经网络·算法·回归·原力计划
蒙奇D索大1 小时前
【数据结构】图论核心算法解析:深度优先搜索(DFS)的纵深遍历与生成树实战指南
数据结构·算法·深度优先·图论·图搜索算法
让我们一起加油好吗1 小时前
【基础算法】高精度(加、减、乘、除)
c++·算法·高精度·洛谷
不会敲代码的灵长类1 小时前
机器学习算法-k-means
算法·机器学习·kmeans
Studying 开龙wu1 小时前
机器学习有监督学习sklearn实战二:六种算法对鸢尾花(Iris)数据集进行分类和特征可视化
学习·算法·机器学习
鑫鑫向栄1 小时前
[蓝桥杯]缩位求和
数据结构·c++·算法·职场和发展·蓝桥杯
Tony__Ferguson1 小时前
简述八大排序(Sort)
数据结构·算法·排序算法
芜湖xin1 小时前
【题解-洛谷】P9422 [蓝桥杯 2023 国 B] 合并数列
算法·队列
鑫鑫向栄1 小时前
[蓝桥杯]外卖店优先级
数据结构·c++·算法·职场和发展·蓝桥杯
<但凡.1 小时前
题海拾贝:P8598 [蓝桥杯 2013 省 AB] 错误票据
数据结构·算法·蓝桥杯