Problem：红与黑

题目描述

有一间长方形的房子，地上铺了红色、黑色两种颜色的正方形瓷砖。你站在其中一块黑色的瓷砖上，只能向相邻（上下左右四个方向）的黑色瓷砖移动。请写一个程序，计算你总共能够到达多少块黑色的瓷砖。

输入格式

输入包括多个数据集合。
每个数据集合的第一行是两个整数 w 和 h，分别表示 x 方向和 y 方向瓷砖的数量。
在接下来的 h 行中，每行包括 w 个字符。每个字符表示一块瓷砖的颜色，规则如下：
1）‘ . ’ ：黑色的瓷砖；
2）‘ # ’ ：红色的瓷砖；
3）‘ @ ’ ：黑色的瓷砖，并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。
当在一行中读入的是两个零时，表示输入结束。

输出格式

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

数据范围

1 ≤ W, H ≤ 20

输入样例

输出样例

思路

1. FloodFill算法
2. DFS与BFS详解.

程序代码

1. BFS写法

#include <iostream>
#include <queue>
#include <algorithm>

using namespace std;

#define x first
#define y second

typedef pair<int, int> PII;
const int N = 25;

int w, h;
char g[N][N];

int bfs(int x, int y) //定义广度优先搜索函数
{
    queue<PII> q; //声明队列q
    q.push({x, y}); //将当前位置压入队列
    g[x][y] = '#'; //置当前位置为已遍历
    int res = 0;

    int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};

    while(q.size())
    {
        auto t = q.front(); //返回队首元素
        q.pop(); //删除队首元素
        res ++;

        for(int i = 0; i < 4; ++ i) //对邻格进行遍历
        {
            x = t.x + dx[i], y  = t.y + dy[i];
            if(x < 0 || x >= h || y < 0 || y >= w || g[x][y] != '.')
                continue;
            g[x][y] = '#';
            q.push({x,y});
        }
    }
    return res;
}

int main()
{
    while(cin >> w >> h, w || h) //w表示列数,h表示行数
    {
        //输入数据集合
        for(int i = 0; i < h; ++ i)
            for(int j = 0; j < w; ++ j)
                cin >> g[i][j];

        //查找当前位置
        int x, y;
        for(int i = 0; i < h; i ++)
        {
            for(int j = 0; j < w; j ++)
                if(g[i][j] == '@')
                {
                    x = i;
                    y = j;
                }
        }

        cout << bfs(x, y) <<endl;//从当前位置开始广度优先搜索并输出结果
    }
    return 0;
}

2. DFS写法

#include <iostream>

using namespace std;

const int N = 25;

int w, h;
char g[N][N];
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};

int dfs(int x, int y) //定义深度优先搜索
{
    int res = 1;
    g[x][y] = '#'; //置当前位置为已遍历
    for(int i = 0; i < 4; ++ i)
    {
        int a = x + dx[i], b = y + dy[i];
        if(a >= 0 && a < h && b >= 0 && b < w && g[a][b] == '.')
            res += dfs(a, b); //递归调用dfs()函数
    }
    return res;//回溯
}

int main()
{
    while(cin >> w >> h, w || h) //w代表列数,h代表行数
    {
        //输入数据集合
        for(int i = 0; i < h; ++ i)
            for(int j = 0; j < w; ++ j)
                cin >> g[i][j];

        //查找当前位置
        int x, y;
        for(int i = 0; i < h; i ++)
        {
            for(int j = 0; j < w; j ++)
                if(g[i][j] == '@')
                {
                    x = i;
                    y = j;
                }
        }

        cout << dfs(x, y) <<endl; //从当前位置开始深度优先搜索并输出结果
    }
    return 0;
}

• 若有问题，欢迎交流