HDU 1241 Oil Deposits（油藏）

题目链接：HDU 1241

题意：

给一个二维字符数组，‘@’代表有油，‘*’代表没有，如果两个油田在水平或竖直或对角线方向相通，那么这两个油田属于同一个pocket，pocket里可能也只有一个油田。问给定的油藏情况共有多少个pocket？

分析：

两天前做的时候还没有头绪，今天拿出来又想了一下，突然有了思路。

同样是BFS搜索。但是每次搜索是从一个尚未被访问的油田开始，然后把从这个油田开始可以连通的所有油田都标记上，也就是搜索这一个pocket里的所有油田。那么在下一次搜索时一定是另一个pocket了。这样一来，有多少次BFS就有多少个pocket了。想通了这一点就很简单了。

注意点：

• 搜索方向有八个，不是4个，也不是6个。
• 标记一定要进队列之前标记，而不是出队列时标记，否则会Memory Limit Exceeded.

CODE:

#include <iostream>
#include <string>
#include <cstring>
#include <queue>
#include <cstdio>
#include <algorithm>
using namespace std;

const int maxn = 110;
int dir[8][2] = { { 0,1 },{ 0,-1 },{ 1,0 },{ -1,0 } ,{-1,-1},{1,1} ,{-1,1},{1,-1} };
int a[maxn][maxn], vis[maxn][maxn], ans, m, n;
char s[maxn];

int valid(int x, int y)
{
	if (x < 0 || y < 0 || x >= m || y >= n || vis[x][y] || a[x][y]==0) return 0;
	return 1;
}
struct Node {
	int x, y;
}cur, nextnode;

void bfs(int w, int r)
{
	ans++;
	queue<Node> q;
	cur.x = w;
	cur.y = r;
	vis[cur.x][cur.y] = 1;
	q.push(cur);
	while (!q.empty())
	{
		cur = q.front();
		q.pop();
		for (int i = 0;i < 8;i++)
		{
			nextnode.x = cur.x + dir[i][0];
			nextnode.y = cur.y + dir[i][1];
			if (!valid(nextnode.x,nextnode.y)) continue;
			vis[nextnode.x][nextnode.y] = 1;
			q.push(nextnode);
		}
	}
	return;
}

int main()
{
#ifdef LOCAL
	freopen("in.txt", "r", stdin);
	freopen("out.txt", "w", stdout);
#endif 
	while (cin >> m >> n)
	{
		if (m == 0) break;
		for (int i = 0;i < m;i++)
		{
			cin >> s;
			for (int j = 0;j < n;j++)
				a[i][j] =( s[j] == '*') ? 0 : 1;
		}
		memset(vis, 0, sizeof(vis));
		ans = 0;
		for (int i = 0;i < m;i++)
			for (int j = 0;j < n;j++)
			{
				if (a[i][j] && !vis[i][j])
					bfs(i, j);
			}
		cout << ans << endl;
	}
	return 0;
}

DFS的写法：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <string>
#include <cmath>
#include <memory>

using namespace std;

const int maxn=110;
char pic[maxn][maxn];
int vis[maxn][maxn],ans,n,m;
int dir[8][2]={{-1,-1},{-1,0},{-1,1},{0,-1},{0,1},{1,-1},{1,0},{1,1}};

void dfs(int r,int c)
{
    for(int i=0;i<8;i++)
    {
        int x=r+dir[i][0];
        int y=c+dir[i][1];
        if(x<1||x>n||y<1||y>m) continue;
        if(pic[x][y]=='@'&&!vis[x][y])
        {
            vis[x][y]=1;
            dfs(x,y);
        }
    }
}

int main()
{
#ifdef LOCAL
    freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
#endif
    while(cin>>n>>m)
    {
        if(n==0) break;
        for(int i=1;i<=n;i++)
            scanf("%s",pic[i]+1);//注意dfs时行列下表都是从1开始的，所以要+1
        memset(vis,0,sizeof(vis));
        ans=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=m;j++)
            {
                if(pic[i][j]=='@'&&vis[i][j]==0)
                {
                    vis[i][j]=1;
                    dfs(i,j);
                    ans++;
                }
            }
        }
        cout<<ans<<endl;
    }
    return 0;
}