HDU 6113 度度熊的01世界【搜索】

题目链接

题意：给个图，判断这个是0还是1还是都不是。

根据条件对0和1分别跑一边DFS求连通块就可以了，非常简单……

WA在跑0的DFS判断是否到边界的时候遇到边界就return了，应该找完所有的0。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <string>
#include <cmath>
#include <vector>
#include <queue>
#include <set>
using namespace std;
#define ll long long

int n,m;
int map[110][110];
bool vis[110][110];

int flag=1;

int dir[10][5]={{0,1},{-1,0},{0,-1},{1,0}};

void dfs1(int x,int y){
    for (int i=0;i<4;i++){
        int tx=x+dir[i][0];
        int ty=y+dir[i][1];
        if (tx<1||tx>n||ty<1||ty>m)
            continue;
        if (map[tx][ty]==1&&!vis[tx][ty]){
            vis[tx][ty]=true;
            dfs1(tx,ty);
        }
    }
}

void find0(int x,int y){
    if (x==1||x==n||y==1||y==m){
        flag=1;
    }
    for (int i=0;i<4;i++){
        int tx=x+dir[i][0];
        int ty=y+dir[i][1];
        if (tx<1||tx>n||ty<1||ty>m)
            continue;
        if (map[tx][ty]==0&&!vis[tx][ty]){
            vis[tx][ty]=true;
            find0(tx,ty);
        }
    }
}

int main(){
    // freopen("1.txt","r",stdin);
    while (scanf("%d %d",&n,&m)!=EOF){
        memset(vis,false,sizeof(vis));
        int x1=0,y1=0;
        for (int i=1;i<=n;i++){
            for (int j=1;j<=m;j++){
                char ch;
                scanf(" %c",&ch);
                map[i][j]=ch-'0';
                if (map[i][j]==1){
                    x1=i;
                    y1=j;
                }
            }
        }
        if (x1==0&&y1==0){ // 如果没有1
            printf("-1\n");
            continue;
        }
        vis[x1][y1]=true;
        dfs1(x1,y1); // 如果dfs1后图中的1都是vis的，则存在1且1由一个连通块组成
        // for (int i=1;i<=n;i++){
        //     for (int j=1;j<=m;j++){
        //         printf("%d",vis[i][j]);
        //     }
        //     putchar('\n');
        // }
        flag=0;
        for (int i=1;i<=n;i++){
            for (int j=1;j<=m;j++){
                if (map[i][j]==1&&!vis[i][j]){
                    flag=1;
                    break;
                }
            }
            if (flag)
                break;
        }       
        if (flag){ // 存在1不在这个连通块中，不满足条件
            printf("-1\n");
            continue;
        }
        int cnt=0;
        for (int i=1;i<=n;i++){
            for (int j=1;j<=m;j++){
                if (map[i][j]==0&&!vis[i][j]){
                    vis[i][j]=true;
                    flag=0;
                    find0(i,j);
                    if (flag==0)
                        cnt++;
                }
            }
        }
        // cout<<"cnt="<<cnt<<endl;
        if (cnt==1){
            printf("0\n");
            continue;
        }
        if (cnt==0){
            printf("1\n");
            continue;
        }
        printf("-1\n");
    }
}