【DFS】数池塘

【DFS】数池塘_第1张图片

题解:
一开始用的BFS做,可能条件判断有一些小问题,只拿了70分。那么这里用的是DFS
遍历8个方向,递归完成后vis是不用还原成0的,我之前的思路是将'W'变为'.'。

每做完一次kcount,那么数量就加一。

代码:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define MAXA 200
using namespace std;
int N,M,cnt;
int direction[10][5] = {{-1,-1},{-1,0},{-1,1},{0,1},{1,1},{1,0},{1,-1},{0,-1}};
char farm[MAXA][MAXA];
int vis[MAXA][MAXA]; 
void kcount(int n,int m)
{
	if(n < 1 || n > N || m < 1 || m > M || farm[n][m] != 'W' || vis[n][m] == 1)
	   return;
	vis[n][m] = 1;
	for(int i=0;i<=7;i++)
	{
		int dx = n + direction[i][0];
		int dy = m + direction[i][1];
		if(dx >= 1 && dx <= N && dy >= 1 && dy <= M && farm[dx][dy] == 'W' && !vis[dx][dy])
		   kcount(dx,dy);
	}
	return;
}
int main()
{
	scanf("%d %d",&N,&M);
	for(int i=1;i<=N;i++)
	    for(int j=1;j<=M;j++)
	    cin >> farm[i][j];
	for(int i=1;i<=N;i++)
	    for(int j=1;j<=M;j++)
	        if(farm[i][j] == 'W' && vis[i][j] == 0)
	        {
	        	kcount(i,j);
	        	cnt++;
			}
	printf("%d",cnt);
}

你可能感兴趣的:(【DFS】数池塘)