poj1562 Oil Deposits（DFS）

题目链接

http://poj.org/problem?id=1562

题意

输入一个m行n列的棋盘，棋盘上每个位置为'*'或者'@'，求'@'的连通块有几个（连通为8连通，即上下左右，两条对角线）。

思路

floodfill问题，用dfs解决。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int N = 110;
char grid[N][N];
int visit[N][N];
int dir[8][2] = {   //八个邻接点
                    {1, 0}, {-1, 0}, {0, 1}, {0, -1},
                    {-1, -1}, {-1, 1}, {1, 1}, {1, -1}
                };
int m, n;

void dfs(int r, int c)
{
    for(int i=0; i<8; i++)
    {
        int nr = r+dir[i][0];
        int nc = c+dir[i][1];

        if(nr>=0 && nr<m && nc>=0 && nc<n && grid[nr][nc]=='@' && !visit[nr][nc])
        {
            visit[nr][nc] = 1;
            dfs(nr, nc);
        }
    }
}

int main()
{
    //freopen("poj1562.txt", "r", stdin);
    while(cin>>m>>n && m)
    {
        for(int i=0; i<m; i++)
            cin>>grid[i];

        memset(visit, 0, sizeof(visit));
        int cnt = 0;
        for(int i=0; i<m; i++)
        {
            for(int j=0; j<n; j++)
            {
                if(grid[i][j]=='@' && !visit[i][j])
                {
                    visit[i][j] = 1;
                    dfs(i, j);
                    cnt++;
                }
            }
        }
        cout<<cnt<<endl;
    }
    return 0;
}