Blocks 题解

题目

为了增强幼儿园小朋友的数数能力,小虎的老师给了一个家庭游戏作业。让小虎拿一块空的围棋盘,随机的在一些方格中放些棋子(有黑白两种颜色),如果一个方格和它的上、下、左、右四个方格之一都有相同颜色的棋子,则认为两格子是相连通的。这期间,要求小虎不断统计共有多少个连通块。

  如下图是一个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
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

你可能感兴趣的:(搜索,c++,算法)