fzu 2150 暴力bfs枚举+stl

点击打开链接

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
using namespace std;
struct node{
int p,q;
node(int p,int q):p(p),q(q) {}
};
const int maxn=10+5;
char maze[maxn][maxn];
bool done[maxn][maxn];
int dist[maxn][maxn];
int n,m;
int dx[]={-1,1,0,0};
int dy[]={0,0,-1,1};
vector<node>str;
bool check()
{
    for(int i=0;i<n;i++)
        for(int j=0;j<m;j++)
              if(maze[i][j]=='#'&&!done[i][j]) return false;
    return true;
}
void dfs(int x,int y)
{
    done[x][y]=true;
    for(int i=0;i<4;i++)
    {
        int nx=x+dx[i],ny=y+dy[i];
        if(nx>=0&&nx<n&&ny>=0&&ny<m&&maze[nx][ny]=='#'&&!done[nx][ny]) dfs(nx,ny);
    }
}
int bfs(int a,int b,int c,int d)
{
    memset(dist,-1,sizeof(dist));
    dist[a][b]=dist[c][d]=0;
    memset(done,false,sizeof(done));
    done[a][b]=done[c][d]=true;
    queue<node>pq;
    while(!pq.empty()) pq.pop();
    int time;
    pq.push(node(a,b));
    if(a!=c||b!=d) pq.push(node(c,d));
    while(!pq.empty())
    {
        node k=pq.front();
        pq.pop();
        time=dist[k.p][k.q];
        done[k.p][k.q]=true;
        for(int i=0;i<4;i++)
        {
            int nx=k.p+dx[i],ny=k.q+dy[i];
            if(nx>=0&&nx<n&&ny>=0&&ny<m&&maze[nx][ny]=='#'&&dist[nx][ny]==-1) {
                dist[nx][ny]=dist[k.p][k.q]+1;
                pq.push(node(nx,ny));
            }
        }
    }
    return time;
}
int main()
{
    int T,kase=0;
    scanf("%d",&T);
    while(T--)
    {
        str.clear();
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
        {
            scanf("%s",&maze[i]);
            for(int j=0;j<m;j++)
                if(maze[i][j]=='#') str.push_back(node(i,j));
        }
        //先求连通分量个数
        memset(done,false,sizeof(done));
        int ans=0;
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
                 if(maze[i][j]=='#'&&!done[i][j]) {
                    ans++;
                    dfs(i,j);
                 }
        printf("Case %d: ",++kase);
        if(ans>2) printf("-1\n");
        else {
                int f=1e6+5;
                for(int i=0;i<str.size();i++)
                    for(int j=i;j<str.size();j++)
                         {
                             int k=bfs(str[i].p,str[i].q,str[j].p,str[j].q);
                             if(check()) f=min(f,k);
                         }
                printf("%d\n",f);
        }
    }
    return 0;
}