poj3026——Borg Maze

题目大意：给出一个y行x列的迷宫，从起点S出发，求出要达到全部的终点A所经过的路径长度最小值，重复走过的路径不算入长度

输入：（有N个case）

             N

             第i个case的列数x 行数y

             第j行迷宫（共y行，#代表墙）

输出：最短路径

分析：最小生成树的应用。任意两点的最小距离就是横坐标之差加上纵坐标之差，但是由于迷宫有墙，所以不能简单的直接求出两点间的距离，而是要引入bfs来计算各点间的距离。然后就套用prim算法求出最小生成树的总路径长度即可。

代码：转载自https://www.cnblogs.com/zhengguiping--9876/p/4684325.html

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

#define N 105
#define INF 0xfffffff
int dir[4][2] = { {1, 0}, {-1, 0}, {0, 1}, {0, -1} };
int D[N][N], n, m, vis[N], used[N][N],b[N][N], dist[N];
char map[N][N];

struct node  //每个非#都是一个结点
{
    int x, y, step;
} a[N];

void bfs(int s, int x, int y)
{
    memset(used, 0, sizeof(used));
    node p,q;
    queue<node>Q;
    p.x = x;
    p.y = y;
    p.step = 0;
    Q.push(p);
    used[x][y] = 1;
    while(Q.size())
    {
        p = Q.front();
        Q.pop();
        if(map[p.x][p.y] >= 'A' && map[p.x][p.y] <= 'Z')
            D[s][ b[p.x][p.y] ] = p.step;
        for(int i=0; i<4; i++)
        {
            q.x = p.x + dir[i][0];
            q.y = p.y + dir[i][1];
            if(q.x<m && q.x>=0 && q.y>=0 && q.y<n && used[q.x][q.y]!=1 && map[q.x][q.y] != '#')
            {
                used[q.x][q.y] = 1;
                q.step = p.step + 1;
                Q.push(q);
            }
        }
    }
}

int Prim(int cnt){  //cnt为图中结点个数
    int ans=0;
    vis[1] = 1;
    for(int i=1; i<=cnt; i++)
        dist[i] = D[1][i];
    for(int i=1; i<=cnt; i++){
        int Min = INF, index = -1;
        for(int j=1; j<=cnt; j++)
            if(vis[j]==0 && Min > dist[j])
            {
                Min = dist[j];
                index = j;
            }
        if(index==-1)break;
        ans += Min;
        vis[index] = 1;
        for(int j=1; j<=cnt; j++)
        {
            if(vis[j] == 0 && dist[j] > D[index][j])
               dist[j] = D[index][j];
        }
    }
    return ans;
}

int main()
{
    int T, ans;
    scanf("%d", &T);
    while(T--)
    {
        memset(a, 0, sizeof(a)); //结点数组初始化
        memset(vis, 0, sizeof(vis));

        memset(map, 0, sizeof(map)); //char型二维矩阵
        for(int i=0; i<N; i++)
        {
            dist[i] = INF;
            for(int j=0; j<N; j++)  //D[i][j]表示编号为i和j的结点的距离
                D[i][j] = INF;
            D[i][i] = 0;
        }

        int cnt = 1;
        scanf("%d%d ", &n,&m);
        for(int i=0; i<m; i++)
        {
            gets(map[i]);
            for(int j=0; j<n; j++)
            {
                if(map[i][j] <= 'Z' && map[i][j] >= 'A')
                    b[i][j] = cnt, cnt++;  //b[i][j]表示map[i][j]位置的结点的编号
            }
        }
        for(int i=0; i<m; i++)
        {
            for(int j=0; j<n; j++)
            {
                if(map[i][j] <= 'Z' && map[i][j] >= 'A')
                    bfs(b[i][j], i, j);  //计算出所有S、A结点到其他每个结点的距离
            }
        }
        ans = Prim(cnt-1);   //cnt-1为迷宫中S和A的总个数
        printf("%d\n", ans);
    }
    return 0;
}