poj 3026 Borg Maze (bfs+prim)

题意：在迷宫中找存在的外星人A撒。。开始是一个组织找，后来这个组织可以分裂成

            很多组。求找到所有外星人的最小步数。（分裂后，步数计算为所有组织的步数和）

思路：因为要保证每个外星人A都找到，加上组织所在起始点S，求最小步数。即求图中

           n个点的最小生成树（n=外星人的个数+1）。

            bfs找每两点间的距离。

1、这题要注意时间复杂度。

开始预处理时把所有外星人和组织起点横纵坐标记录后。用了两个for循环，来找每两点的

距离。这样要调用n*n次bfs去搜状态，结果tle。其实每次重复搜没必要。

将n*n复杂度可以降到n。也就是遍历n个点，以其中每个点作起点一次，来搜到达其他各点

的最短距离。

2、这题要注意数组大小

外星人最多100，加上组织起点S一共101个点。

数组开小就wa了。

3、这题读入迷宫行列后不能用getchar，因为后面不止是单纯的回车。还有多余空格。

要用gets()吃掉所有。

代码：

#include<cstdio>#include<iostream>#include<cstring>#include<cmath>#include<cstdlib>#include<queue>#define INF 0x3f3f3f3fusing namespace std;struct node{    int x,y;    int step;};char m[55][55];int r[110],c[110],ans,rear,x,y;int edge[110][110],dis[110];bool vis[110];bool v[55][55];int dx[]={-1,1,0,0};int dy[]={0,0,-1,1};queue<node> Q;void dist(int i){    while(!Q.empty())        Q.pop();    node a,b,tmp;    int num=0;    a.x=r[i],a.y=c[i];    a.step=0;    Q.push(a);    v[r[i]][c[i]]=1;    while(!Q.empty())    {        tmp=Q.front();        Q.pop();        for(int k=0;k<rear;k++)        {            if(tmp.x==r[k] && tmp.y==c[k])            {                edge[k][i]=edge[i][k]=tmp.step;                num++;                break;            }        }        if(num==rear)            break;        for(int k=0;k<4;k++)        {            b.x=tmp.x+dx[k];            b.y=tmp.y+dy[k];            b.step = tmp.step+1;            if(b.x>=1 && b.x<=y && b.y>=1&&b.y<=x && m[b.x][b.y]!='#'&& !v[b.x][b.y])            {                v[b.x][b.y]=1;                Q.push(b);            }        }    }}void prim(){    int u=0;    for(int i=1;i<rear;i++)    {        dis[i]=edge[u][i];    }    vis[u]=1;    for(int i=1;i<rear;i++)    {        int minc = INF;        for(int j=0;j<rear;j++)        {            if(!vis[j] && dis[j]<minc)            {                u=j;                minc=dis[j];            }        }        ans+=minc;        vis[u]=1;        for(int j=0;j<rear;j++)        {            if(!vis[j] && edge[u][j]<dis[j])                dis[j]=edge[u][j];        }    }}int main(){    int T,sx,sy;    char shit[50];    scanf("%d",&T);    while(T--)    {        memset(vis,0,sizeof(vis));        scanf("%d%d",&x,&y);        gets(shit);        for(int i=1;i<=y;i++)            gets(m[i]+1);        rear=0;        for(int i=1;i<=y;i++)        {            for(int j=1;j<=x;j++)            {                if(m[i][j]=='S')                {                    sx=i;                    sy=j;                }                else if(m[i][j]=='A')                {                    r[rear]=i;                    c[rear++]=j;                }            }        }        r[rear]=sx;        c[rear++]=sy;        for(int i=0;i<rear;i++)        {            memset(v,0,sizeof(v));            dist(i);        }        ans=0;        prim();        printf("%d\n",ans);    }    return 0;}