POJ3026-Borg Maze（Kruskal+bfs）

题目描述

Input

Output

Sample Input

2
6 5
##### 
#A#A##
# # A#
#S  ##
##### 
7 7
#####  
#AAA###
#    A#
# S ###
#     #
#AAA###
#####  

Sample Output

8
11

题意

t组提问，每组n*m个点，S代表起点，A代表外星人，#表示边界，搜索到外星人后可以继续移动，求最小移动的步数

思路

最小生成树+BFS，BFS求每个点到其它点最短距离（因为有墙，不能直接用坐标求曼哈顿距离），然后套最小生成树的模板

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
//Kruskal

using namespace std;
typedef long long ll;
const int MAXN =4e5+10;
const int N = 5e2+10;
char a[N][N],ssr[N];//a存地图
int g[N][N]= {0};//表示点是否走过
int d[N][N]= {0};//表示到其他点的距离
int ha[N][N];//表示点的下标（第几个点）
struct node//边
{
    int x,y,z;
} e[MAXN];
struct pos//坐标
{
    int x,y;
} p[MAXN];
bool cmp(node a,node b)
{
    return a.z<b.z;
}
int pre[MAXN];
int findd(int x)//找父节点
{
    if(x!=pre[x])
    {
        int t=findd(pre[x]);
        pre[x]=t;
    }
    return pre[x];
}
void unity(int x,int y)
{
    int fx = findd(x), fy = findd(y);
    if (fx != fy)
    {
        pre[fy] = fx;
    }
}
int judge(int x,int y)//判断x,y是否连通(父节点相同)
{
    int fx=findd(x),fy=findd(y);
    if(fx!=fy)return 1;
    return 0;
}
int x1[]= {1,0,0,-1};
int y1[]= {0,1,-1,0};
int n,m;
int nbnum=1;//点的下标
int num_edge=1;//边数
void bfs(int x,int y)
{
    if(!ha[x][y])
        ha[x][y]=nbnum;
    memset(g,0,sizeof(g));
    memset(d,0,sizeof(d));
    queue<pos>q;
    q.push({x,y});
    while(!q.empty())
    {
        pos st=q.front();
        q.pop();
        for(int i=0; i<4; i++)
        {
            int xx=st.x+x1[i];
            int yy=st.y+y1[i];
            if(g[xx][yy]!=1&&xx>=1&&yy>=1&&xx<=n&&yy<=m)//在边界内
           {
                if(a[xx][yy]!='#')
                {
                    g[xx][yy]=1;//标记走过
                    d[xx][yy]=d[st.x][st.y]+1;//距离+1
                    q.push({xx,yy});
                    if(a[xx][yy]=='A'||a[xx][yy]=='S')//记录点坐标（S和A可以看作相同类型）
                    {
                        if(ha[xx][yy]==0)
                            ha[xx][yy]= ++nbnum;
                        if(ha[x][y]!=ha[xx][yy])
                        {
                            e[num_edge].x=ha[x][y];
                            e[num_edge].y=ha[xx][yy];
                            e[num_edge].z=d[xx][yy];
                            num_edge++;
                        }
                    }
                }
            }
        }
    }
}

int main()
{
    int ans;
    int t;
    scanf("%d",&t);
    while(t--)
    {
        nbnum=1;
        num_edge=1;
        memset(ha,0,sizeof(ha));
        memset(g,0,sizeof(g));
        memset(e,0,sizeof(e));
        scanf("%d%d",&m,&n);
        ans=0;
        gets(ssr);//这个B输入多了一堆空格，用一个字符串吃掉
        for(int i=1; i<=n; i++)
        {
            gets(a[i]);
        }
        ans=0;
        for(int i=1; i<=n; i++)
            for(int j=1; j<=m; j++)
            {
                if(a[i][j]=='S'||a[i][j]=='A')
                {
                    bfs(i,j);
                }
            }
        for(int i=0; i<=nbnum; i++)
            pre[i]=i;
        sort(e+1,e+1+num_edge,cmp);
        //cout<<num_edge<<endl<<endl;
        //for(int i=0; i<=num_edge+1; i++)cout<<e[i].x<<" "<<e[i].y<<" "<<e[i].z<<endl;
        for(int i=0; i<=num_edge+1; i++)
        {
            if(judge(e[i].x,e[i].y))
            {
                unity(e[i].x,e[i].y);
                ans+=e[i].z;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}

莫名其妙WA了好几发，输入有坑，数字后面跟了一堆空格，离谱