欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【acwing 寒假每日一题】day4红与黑

程序员文章站 2022-03-06 19:05:58
...

题目来源https://www.acwing.com/problem/content/1115/

题目描述

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

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

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

输入格式
输入包括多个数据集合。

每个数据集合的第一行是两个整数 W 和 H,分别表示 x 方向和 y 方向瓷砖的数量。

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

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

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

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

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

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

输出样例:

45

思路

问题理解:就是求人所在的由 . 构成的连通块的数量
Flood Fill算法:
针对网格图搜索
可以采用dfs(更方便)或者bfs(可以求得最短路)实现

代码

bfs

#include<bits/stdc++.h>

using namespace std;

//简化代码 pair的.first .second 就缩略成.x .y
#define x first
#define y second
typedef pair<int,int> PII;

const int N=25;
char g[N][N];
int n,m;

int bfs(int sx,int sy)
{
    queue<PII> q;
    q.push({sx,sy});
    g[sx][sy]='#';
    
    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++)
        {
            int x=t.x+dx[i];
            int y=t.y+dy[i];
            //排除非法情况
            if(x>=0 && x<n && y>=0 && y<m && g[x][y]=='.')
            {
                g[x][y]='#';
                q.push({x,y});   
            }
        }
    }
    return res;
}

int main()
{
    //读入多个数据集合
    while(cin>>m>>n,m || n)
    {
        //读入每一组数据
        for(int i=0;i<n;i++) cin>>g[i];
        //找到起始位置
        int x,y;
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
                if(g[i][j]=='@')
                {
                    x=i;
                    y=j;
                }
        //从起始位置开始拓展
        cout<<bfs(x,y)<<endl;
    }
    return 0;
}

dfs

#include<bits/stdc++.h>

using namespace std;

const int N=25;
char g[N][N];
int n,m;

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<n && b>=0 && b<m && g[a][b]=='.')
        {
            res+=dfs(a,b);
        }
    }
    return res;
}


int main()
{
    //读入多个数据集合
    while(cin>>m>>n,m || n)
    {
        //读入每一组数据 因为输入样例每行中的字符之间没有空格 所以直接cin就行了
        for(int i=0;i<n;i++) cin>>g[i];
        //找到起始位置
        int x,y;
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
                if(g[i][j]=='@')
                {
                    x=i;
                    y=j;
                }
        //从起始位置开始拓展
        cout<<dfs(x,y)<<endl;
    }
    return 0;
}