上次分享了我在白书当中学习到的DFS,使用了递归的思想。DFS本身是一个栈式数据结构,对于图,只是将其转化为数据进行操作。并没有使用Stack,但是递归本身就是将局部变量存储到栈帧,当中。对于很大的数据范围,栈容易发生溢出,导致出错。下面用手写栈代替递归调用。
代码如下:
#include <stdio.h>
#include <memory.h>
int mat[110][110], vis[110][110];
int sta[10010];//使用数组开辟一个栈
int dx[8] = {-1, -1, -1, 0, 0, 1, 1, 1}, dy[8] = {-1, 0, 1, -1, 1, -1, 0, 1};//偏移量
void dfs(int x, int y){
int top = -1, i, nx, ny;//头指针初始化为-1相当重要,对于判别栈是否为空概念要清晰
int u = x*1000 + y;//u压缩保存x、y的值
sta[++top] = u;//起点进栈
while(top != -1){//在栈为空之前,按一定顺序访问结点,直到按条件完成一次dfs
for(i = 0; i < 8; i++){//朝八个方向偏移
nx = u / 1000 + dx[i];
ny = u % 1000 + dy[i];
if(mat[nx][ny]&& !vis[nx][ny]){
vis[nx][ny] = 1;
u = sta[++top] = nx*1000 + ny;//访问成功,当前位置进栈,u更新为当前位置
break;//一旦访问成功,则不再访问其他偏移方向。(先进后出)
}
}
if(i == 8)
u = sta[--top];//数到山穷水尽,方知此湖到岸,栈头出栈,u更新为新栈头
}
}
int main(){
int i, j, cou = 0;
int l, r;
scanf("%d%d", &l, &r);
char s[r+1];
memset(mat, 0, sizeof(mat));
memset(vis, 0, sizeof(vis));
for(i = 0; i < l; i++){
scanf("%s", s);
for(j = 0; j < r; j++)
mat[i+1][j+1] = (s[j] == 'W' ? 1 : 0);
}
for(i = 1; i <= l; i++){
for(j = 1; j <= r; j++){
if(!vis[i][j] && mat[i][j]){
cou++;
dfs(i, j);
}
}
}
printf("%d\n", cou);
return 0;
}