AcWing 1113. 红与黑 （BFS 、DFS)

AcWing 1113. 红与黑

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

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

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

数据范围
1≤W,H≤20
输入样例：

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

输出样例：

45

方法1：BFS
AC代码

#include<bits/stdc++.h>
using namespace std;
const int N=50;
typedef pair<int,int>PII; //pair用来存坐标较为方便
char g[N][N];
int w,h; //要注意，w是列，h是行
int bfs(PII start)
{
    queue<PII>q; //定义一个队列
    q.push(start);//入队
    int cnt=1; //统计数量
    while(!q.empty())
    {
        PII t=q.front();
        q.pop(); //队头出队
        // 定义四个方向
        int dx[4]={-1,0,1,0};
        int dy[4]={0,1,0,-1};
        for(int i=0;i<4;i++)
        {
            int x=t.first+dx[i];
            int y=t.second+dy[i];
            if (x<0||x>=w||y<0||y>=h) continue;  // 出界
            if(g[x][y]=='#') continue; //不可移动
            if(g[x][y]=='.') 
            {
                 g[x][y]='#';//标记为不可移动
                 q.push({x,y}); //坐标入队
                 cnt++; //数量 +1
            }
        } 
    } return cnt;
}
int main()
{
    while(cin>>h>>w,w||h)//当在一行中读入的是两个零时，表示输入结束
    {
        PII start; //定义起点
        memset(g,'#',sizeof g); //初始化
        for(int i=0;i<w;i++)
        for(int j=0;j<h;j++)
        {
            cin>>g[i][j];
            if(g[i][j]=='@')//找到起点坐标
            {
                start.first=i;
                start.second=j;
                g[i][j]='#'; //标记为不可移动位置
            }
        }
        cout<<bfs(start)<<endl;; 
    }
    return 0;
}

方法2：DFS
AC代码

#include<bits/stdc++.h>
using namespace std;
const int N=25;
char g[N][N];
int w,h; //w是列，h是行
int cnt;
int dfs(int x,int y)
{
    g[x][y]='#'; //标记为红色
    int cnt=1; //每次符合条件递归 数量 +1
    //定义四个方向
    int dx[4]={-1,0,1,0};
    int dy[4]={0,1,0,-1};
    for(int i=0;i<4;i++)
    {
        int xx=x+dx[i];
        int yy=y+dy[i];
        if(g[xx][yy]=='.')
        {
            cnt+=dfs(xx,yy); //递归
        }
    }
    return cnt;
}
int main()
{
    //先读列，再读行，要注意注意注意！！！
    while(cin>>w>>h,w||h)//当在一行中读入的是两个零时，表示输入结束
    {
        int x,y; //记录起点坐标
        for(int i=0;i<h;i++)
        for(int j=0;j<w;j++)
        {
            cin>>g[i][j];
            if(g[i][j]=='@')//找到起点坐标
            {
                x=i,y=j;
                g[i][j]='#'; //标记为红色
            }
        }
       cout<<dfs(x,y)<<endl;
    }
    return 0;
}

如果觉得写的还不错，那就一键三连吧 ^ ^