题目:输入一个n*n的黑白图像(1表示黑色,0表示白色),任务是统计其中八连块的个数。如果两个黑格子有公共边或者公共顶点,就说它们属于同一个八连块。如图表示3个八连块
- //Depth-First Search
- #include <stdio.h>
- #include <string.h>
- #define MAX 100
- int mat[MAX][MAX], vis[MAX][MAX];
- void dfs(int x, int y)
- {//深度优先搜索
- if (!mat[x][y] || vis[x][y])
- {//曾经访问过这个格子,或者当前格子是白色
- return;
- }
- vis[x][y] = 1; //标记(x, y)已访问过
- //递归访问周围的八个格子
- dfs(x-1, y-1); dfs(x-1, y); dfs(x-1, y+1);
- dfs(x, y-1); dfs(x, y+1);
- dfs(x+1, y-1); dfs(x+1, y); dfs(x+1, y+1);
- }
- int main(void)
- {
- int n, i, j;
- int count;
- char s[MAX];
- memset(mat, 0, sizeof(mat));//将所有的格子都初始化为白色,包括周围一圈虚拟格子
- memset(vis, 0, sizeof(vis));//所有的格子都没有访问过
- scanf("%d", &n);
- for (i=0; i< n; i++)
- {
- memset(s, 0, sizeof(s));
- scanf("%s", s);
- for (j=0; j<n; j++)
- {
- mat[i+1][j+1] = s[j] - '0'; //把图像往中间移动一点,空出一圈白格
- }
- }
- count = 0;
- for (i=1; i<=n; i++)
- {
- for (j=1; j<=n; j++)
- {
- if (!vis[i][j] && mat[i][j])
- {//寻找没有访问过的黑格子
- count++;
- dfs(i, j);
- }
- }
- }
- printf("%d\n", count);
- return 0;
- }
DFS(Depth-First Search) 用到的是栈,递归调用时会把局部变量(当前结点编号)存入栈帧中,如果图像太大,递归方式的DFS有栈溢出的危险。为保险起见,可以用显式栈来代替递归调用。如下
- //Depth-First Search
- #include <stdio.h>
- #include <string.h>
- #define MAX 100
- int mat[MAX][MAX], vis[MAX][MAX];
- int s[MAX];//显式栈的栈应为全局变量,因为局部变量定义在栈里面
- int dx[8] = {-1,-1,-1,0,0,1,1,1};//方向定位
- int dy[8] = {-1,0,1,-1,1,-1,0,1};
- void dfs(int x, int y)
- {//深度优先搜索
- int u = x*8+y;//u代表(x,y)在数组中的编号,即用一个数表示(x,y)的位置
- int top = 0;
- s[top++] = u;//将(x,y)的位置存放在栈中
- vis[x][y] = 1;//访问记录
- while (top)
- {//栈非空时
- u = s[--top];//取出栈顶存储的块的编号
- x = u/8; y = u%8;//将编号还原为坐标
- for (int d=0; d<8; d++)
- {//i代表要走的8个方向
- int nx = x + dx[d], ny = y + dy[d];//(nx, ny)代表下一个访问位置,即访问一个新块
- if (mat[nx][ny] && !vis[nx][ny])
- {//若(nx, ny)为黑色图且没被访问过
- int v = nx * 8 + ny;//获取新块的编号
- vis[nx][ny] = 1;//访问记录
- s[top++] = v;//将新块放入栈
- d = -1;//重新遍历新块的四周
- x = nx; y = ny;//新块编号变成旧块坐标
- }
- }
- }
- }
- int main(void)
- {
- int n, i, j;
- int count;
- char s[MAX];
- memset(mat, 0, sizeof(mat));//将所有的格子都初始化为白色,包括周围一圈虚拟格子
- memset(vis, 0, sizeof(vis));//所有的格子都没有访问过
- scanf("%d", &n);
- for (i=0; i< n; i++)
- {
- memset(s, 0, sizeof(s));
- scanf("%s", s);
- for (j=0; j<n; j++)
- {
- mat[i+1][j+1] = s[j] - '0'; //把图像往中间移动一点,空出一圈白格
- }
- }
- count = 0;
- for (i=1; i<=n; i++)
- {
- for (j=1; j<=n; j++)
- {
- if (!vis[i][j] && mat[i][j])
- {//寻找没有访问过的黑格子
- count++;
- dfs(i, j);
- }
- }
- }
- printf("%d\n", count);
- return 0;
- }