算法标签 DFS，Flood Fill

来源： 《信息学奥赛一本通》

题目描述

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

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

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

输入格式

输入包括多个数据集合。

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

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

1）‘.’：黑色的瓷砖；
2）‘#’：白色的瓷砖；
3）‘@’：黑色的瓷砖，并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。

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

输出格式

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

数据范围

1≤W,H≤20

输入样例：

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

输出样例：

45

思路

深搜代码更短，但层数过多更有可能爆栈。
o(n^2)遍历

C++ 代码

#include<iostream>
#include<cstring>

using namespace std;

const int N=25;
char g[N][N];
int w,h;
bool st[N][N];//状态判断

int dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};//题目中能移动的相邻的四个方向的偏移量

int dfs(int x,int y)
{
    int cnt=1;
    st[x][y]=true;//初始时计算当前次数一次  判断当前位置为走过
    
    for(int i=0;i<4;i++)//四个方向都走一遍
        {
            int a,b;
            a=x+dx[i],b=y+dy[i];
            
            if(a<0||a>w||b<0||b>h)continue;
            if(g[a][b]!='.')continue;
            if(st[a][b])continue;//判断当前位置不可行的条件
            
            cnt+=dfs(a,b);//如果当前位置允许则进入下一层
        }
    
    return cnt;//返回数量
}
int main()
{
    while(cin>>h>>w,w||h)//当在一行中读入的是两个零时停止
        {
            for(int i=0;i<w;i++)cin>>g[i];//读入地图
            
            int x=0,y=0;
            for(int i=0;i<w;i++)
                for(int j=0;j<h;j++)
                    if(g[i][j]=='@')//找到起始点的位置
                        x=i,y=j;
            
            memset(st,0,sizeof st);//在多次状态中都要使用 所以每次都要清空
            cout<<dfs(x,y)<<endl;
        }
    
    return 0;
}