POJ 3026 Borg Maze （最小生成树 + BFS）

将S点和A点看做是相同的（其实也没有任何区别），从S点和每个A点做BFS，算出到其他点的最短距离，再用Prim算法算最小生成树（Prim适合稠密图）。

注意此题输入最好用gets()，测试数据里面有很多多余的空格。

此题主要考代码掌控能力，对多个算法的拼接，考的算法都很基本。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
#include <cmath>
#include <string>
#include <map>
#include <queue>

using namespace std;

char s[111][111];
int g[111][111];
bool v[111];
bool vv[111][111];
int dis[111];
queue<pair<pair<int,int>,int > > q;
int d[4][2] = {0,1,0,-1,1,0,-1,0};

int n,m,num;
struct k
{
    int x,y;
}a[111];
map<int,int> ma;

int prim()
{
    memset(v,0,sizeof(v));
    for(int i=1;i<=num;i++) dis[i] = 100000;
    dis[1] = 0;
    int ans = 0;
    for (int i=1;i<=num ;i++ )
    {
        int mark = -1;
        for (int j = 1;j<=num ;j++ )
        {
            if(!v[j])
                if(mark==-1)
                    mark = j;
                else if(dis[j]<dis[mark])
                    mark = j;
        }
        if(mark == -1)
            break;
        v[mark] = 1;
        ans += dis[mark];
        for(int j=1;j<=num;j++)
            if(!v[j])
            {
                int x = j;
                dis[x] = min(dis[x],g[mark][j]);
            }
    }
    return ans;
}

void bfs(int i)
{
    memset(vv,0,sizeof(vv));
    int x = a[i].x;
    int y = a[i].y;
    int xx,yy,len;
    while(!q.empty()) q.pop();
    q.push(make_pair(make_pair(x,y),0));
    vv[x][y] = true;
    while(!q.empty())
    {

        x = q.front().first.first;
        y = q.front().first.second;
        len = q.front().second;
        if(s[x][y]=='S'|| s[x][y] == 'A')
            g[i+1][ma[111*x+y]+1] = len;
        q.pop();
        for(int i=0;i<4;i++)
        {
            xx = x + d[i][0];
            yy = y + d[i][1];
            if(xx>=0 && xx<=n && yy>=0 && yy<=m && !vv[xx][yy] && s[xx][yy]!='#')
            {
                q.push(make_pair(make_pair(xx,yy),len+1));
                vv[xx][yy] = true;
            }
        }
    }
}

void slove()
{
    num = 0;
    scanf("%d%d ",&m,&n);
    for(int i=0;i<n;i++)
        gets(s[i]);
    for(int i=0;i<n;i++)
        for(int j=0;j<m;j++)
            if(s[i][j] == 'S' || s[i][j] == 'A')
            {
                a[num].x = i;
                a[num].y = j;
                ma[111*i+j] = num++;
            }
    memset(g,0,sizeof(g));
    for(int i=0;i<num;i++)
        bfs(i);
    printf("%d\n",prim());
}

int main()
{
#ifdef ARTHUR_YANG
freopen("in.txt","r",stdin);
#endif // ARTHUR_YANG
    //ios::sync_with_stdio(false);
    int n;
    while(~scanf("%d",&n))
    {
        while(n--)
            slove();
    }
}