题目
为了增强幼儿园小朋友的数数能力,小虎的老师给了一个家庭游戏作业。让小虎拿一块空的围棋盘,随机的在一些方格中放些棋子(有黑白两种颜色),如果一个方格和它的上、下、左、右四个方格之一都有相同颜色的棋子,则认为两格子是相连通的。这期间,要求小虎不断统计共有多少个连通块。
如下图是一个5*9的一块棋盘,其中'.'表示空格,'*'表示黑棋子,'@'表示白棋子。则有4块连通的棋子块。
.........
..**..@..
.**@@.@@.
..*@..*..
.........
哥哥大虎在一边看一边想,如果棋盘是N*N的,共放了M个棋子,如何用计算机解决这个问题呢?
输入格式
第一行两个整数:N M
接下来有M行,每行三个正整数: C X Y (0<=C<=1, 1<=X,Y<=N)。分别表示依次放入棋子的颜色(0表示白色,1表示黑色)、要放入格子的横坐标和格子的纵坐标。
30%数据:1<=N<=10
60%数据:1<=N<=100
100%数据:1<=N<=500 1<=M<=N*N。
输出格式
共M行。第i行一个整数,表示放入第i个棋子后,当前有多少个棋子连通块。
输入/输出例子1
输入:
3 5
1 1 1
1 1 2
0 2 2
1 3 1
1 2 1
输出:
1
1
2
3
2
输入/输出例子2
输入:
3 5
1 1 2
1 2 1
1 3 2
1 2 3
1 2 2
输出:
1
2
3
4
1
这题虽然可以用并查集,但要求用DFS
所以就有了这串代码:
#include<bits/stdc++.h>
using namespace std;
int n,m,opt,l,h,ans=0;
int dx[4]={-1,1,0,0};
int dy[4]={0,0,1,-1};
int vis[509][509],g[509][509];
bool dfs(int x,int y,int state){
if(g[x][y]!=opt) return false;
if(x<1||x>n||y<1||y>n) return false;
if(vis[x][y]==state) return false;
vis[x][y]=state;
for(int j=0;j<4;j++){
int nx=x+dx[j];
int ny=y+dy[j];
dfs(nx,ny,state);
}
return true;
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++){
scanf("%d%d%d",&opt,&l,&h);
opt++;
for(int j=0;j<4;j++){
int nx=l+dx[j];
int ny=h+dy[j];
if(dfs(nx,ny,i)) ans--;
}
ans++;
g[l][h]=opt;
vis[l][h]=i;
printf("%d\n",ans);
}
return 0;
}
最奇怪的是,vis数组为int不为bool,这样不用每次清零vis,
总结
DFS四向搜,遇上类型不同或找过或出界,返回false