Borg Maze POJ - 3026 bfs 加prim算法

/**
坑点就想喷死出题的，，我想喷出题的  不知道坑人要偿命的吗？，坑点出现浪费时间，贡献了很多WA，很是不爽，mmp；
讲一下我理解的题意吧；
就是给你一个n*m的地图，存在S和A两个字母代表村庄，然后空格代表可以走(一个村庄的人到另一个村庄可以走，通路)，然后#代表不能走；
它需要修建马路使得所有的村庄尽可能连接起来，那么要你求修建起这些点所花最短时间；
咋们来分析一下哈，首先它没有给咱现成的图，还有就是咋们应该怎样把它连起来呢；
所以我们需要把每两个点连接起来，找每个点到另一个点的距离；这样就形成了一幅完整的图了；
有图后，它需要把他们所有的点都连接起来，找到一条通路使得所有边值和最小；
所以我们在找距离的时候可以bfs可以找到所有的点，然后再进行prim找到边圈最小；
*/

#include<cstdio>
#include<cmath>
#include<queue>
#include<cstring>
using namespace std;

const int inf=0x3f3f3f3f;
const int maxn=2e3+7;
int mmp[maxn][maxn],n,ans,m,cnt,w,dis[maxn],pos[maxn][maxn];
bool Vis[maxn][maxn],vis[maxn];
char str[maxn][maxn];
int dx[]={0,0,1,-1};
int dy[]={1,-1,0,0};

struct node
{
    int x,y,step;
};

void bfs(int x,int y)
{
    memset(Vis,false,sizeof(Vis));
    queue<node>Q;
    node now,next;
    now.x=x;
    now.y=y;
    now.step=0;
    Vis[x][y]=true;
    Q.push(now);
    while(!Q.empty())
    {
        now=Q.front();
        Q.pop();
        if(str[now.x][now.y]=='A'||str[now.x][now.y]=='S') ///找到一个；就上图车；
            mmp[pos[x][y]][pos[now.x][now.y]]=mmp[pos[now.x][now.y]][pos[x][y]]=now.step;
            for(int i=0;i<4;i++)
            {
                int x1=now.x+dx[i];
                int y1=now.y+dy[i];
                if(x1>=1&&x1<=m&&y1>=1&&y1<=n&&str[x1][y1]!='#'&&!Vis[x1][y1])
                {
                    Vis[x1][y1]=true;
                    next.x=x1;
                    next.y=y1;
                    next.step=now.step+1;
                    Q.push(next);
                }
            }
    }
}

void prim()
{
    ans=0;
    for(int i=1;i<=cnt;i++)
    {
        vis[i]=0;
        dis[i]=mmp[1][i];
    }
    vis[1]=1;
    for(int i=1;i<cnt;i++)
    {
        int  Min=inf,k=-1;
        for(int j=1;j<=cnt;j++)
        {
            if(!vis[j]&&dis[j]<Min)
            {
                Min=dis[j];
                k=j;
            }
        }
        if(k==-1) break;
        ans+=Min;
        vis[k]=1;
        for(int j=1;j<=cnt;j++)
            if(!vis[j]&&dis[j]>mmp[k][j]) dis[j]=mmp[k][j];
    }
    printf("%d\n",ans);
}

int main ()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d\n",&n,&m);
        /**
        这踏马不知道m的后面有多少个空格；
        我还特意去看了一下poj的discuss  全是在喷出题的，坑点浪费时间，毫无意义；
        不过个人感觉还是挺好的，长见识了，喷完瞬间就舒服了；
        */
        ///两种写法；借鉴一波；
            /**cin>>x>>y;
            char c  = getchar();
            while(c != '\n')
            {
                c = getchar();
            }*/

        cnt=0;
        for(int i=1;i<=m;i++)
        {
            gets(str[i]);
            for(int j=0;j<n;j++)
            {
                if(str[i][j]=='S'||str[i][j]=='A') pos[i][j]=++cnt;///查找它节点的个数方便建图；
            }
        }

        for(int i=1;i<=cnt;i++)
            for(int j=1;j<=cnt;j++)
            if(i==j) mmp[i][j]=0;
        else mmp[i][j]=inf;

        for(int i=1;i<=m;i++)
            for(int j=1;j<=n;j++)
                if(str[i][j]=='S'||str[i][j]=='A') bfs(i,j);///找到它的节点，顺着它进行查找,顺便进行建图；

                prim();
    }
    return 0;
}