红与黑

_{铀是一道水题}

有一个矩形房间，覆盖正方形瓷砖。每块瓷砖涂成了红色或黑色。一名男子站在黑色的瓷砖上，由此出发，可以移到四个相邻瓷砖之一，但他不能移动到红砖上，只能移动到黑砖上。编写一个程序，计算他通过重复上述移动所能经过的黑砖数（站立的黑砖也要算）。

输入

　　开头行包含两个正整数W和H，W和H分别表示矩形房间的列数和行数，且都不超过20.
　　每个数据集有H行，其中每行包含W个字符。每个字符的含义如下所示：
　　'.'——黑砖
　　'#'——红砖
　　'@'——男子（仅出现一次）

输出

　　男子从初始瓷砖出发可到达的瓷砖数

样例

样例输入1

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

样例输出1

45

 

code:

#include <bits/stdc++.h>
using namespace std;
char mp[21][21];
int w,h,vis[21][21],cnt;
int a[4]={1,0,-1,0};
int b[4]={0,1,0,-1}; 
struct zb
{
    int x,y;
}s;
queue<zb> q;

void bfs()
{
    zb now,nt;
    q.push(s);
    while(q.size())
    {
        now=q.front();
        q.pop();
        
        for(int i=0;i<4;i++)
        {
            nt.x = now.x+a[i], nt.y = now.y+b[i];
            if(!vis[nt.x][nt.y] && mp[nt.x][nt.y]=='.' && nt.x>=0 && nt.x<h && nt.y>=0 && nt.y<w)
            {
                cnt++;
                vis[nt.x][nt.y]=1;
                q.push(nt);
            }
        } 
    }
}

int main()
{

    cin >> w >> h;
    cnt=1;
    for(int i=0;i<h;i++)
    {
        for(int j=0;j<w;j++)
        {
            cin >> mp[i][j];
            if(mp[i][j]=='@')
            {
                s.x=i;
                s.y=j;
            }
        }
    }
    bfs();
    cout<<cnt;
    return 0;
}