Flood Fill算法

Flood Fill算法

Flood Fill算法又叫洪水填充算法,一般用于图像处理。用于题目上的话一般就是海上的陆地一类的题目。

这里以本题为例对Flood Fill算法进行讲解。


题目描述

有一间长方形的房子,地上铺了红色、黑色两种颜色的正方形瓷砖。

你站在其中一块黑色的瓷砖上,只能向相邻(上下左右四个方向)的黑色瓷砖移动。

请写一个程序,计算你总共能够到达多少块黑色的瓷砖。

输入格式

输入包括多个数据集合。

每个数据集合的第一行是两个整数 WW 和 HH,分别表示 xx 方向和 yy 方向瓷砖的数量。

在接下来的 HH 行中,每行包括 WW 个字符。每个字符表示一块瓷砖的颜色,规则如下

1)‘.’:黑色的瓷砖;
2)‘#’:红色的瓷砖;
3)‘@’:黑色的瓷砖,并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。

当在一行中读入的是两个零时,表示输入结束。

输出格式

对每个数据集合,分别输出一行,显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。

数据范围

1≤W,H≤201≤W,H≤20

输入样例:

6 9 
....#. 
.....# 
...... 
...... 
...... 
...... 
...... 
#@...# 
.#..#. 
0 0

输出样例:

45

本题我们采用DFS写法,具体流程如下:

By zy

伪代码:

int dfs(int x, int y) {
    标记此位置为已走过的点
    枚举四个偏移量
    if(邻格可走) {
        dfs(邻格坐标);
    }
}

代码实现:

#include 

using namespace std;

const int N = 25;
int m, n;
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
char arr[N][N];


int dfs(int x, int y) {
    arr[x][y] = '#';
    int res = 1;
    for(int i = 0; i < 4; i++) {
        int a = x + dx[i];
        int b = y + dy[i];
        if(a >= 0 && a < n && b >= 0 && b < m && arr[a][b] == '.') {
            res += dfs(a, b);
        }
    }
    return res;
}

int main() {
    while(cin >> m >> n, n || m) {
        for(int i = 0; i < n; i++) {
            cin >> arr[i];
        }
        int x, y;
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < m; j++) {
                if(arr[i][j] == '@') {
                    x = i;
                    y = j;
                }
            }
        }
        cout << dfs(x, y) << endl;
    }
    return 0;
}

这题还是用到了偏移法,即dx[] = {-1, 0, 1, 0}dy[] = {0, 1, 0, -1},这两个偏移数组可以记住,顺序是上、右、下、左。