Borg Maze---poj3026最小生成树+bfs

题目链接：http://poj.org/problem?id=3026

题意：可以理解为给你一个地图上面有S和A的表示一个一个的点,#表示墙，所以需要我们用bfs来计算各点之间的距离；

b[i][j] = p;代表map[i][j]的位置的点的编号为p；

D[i][j] 代表编号为i和j的距离；

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<iostream>
using namespace std;

#define N 105
#define INF 0xfffffff
int dir[][] = { {, }, {-, }, {, }, {, -} };
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, , sizeof(used));
    node p,q;
    queue<node>Q;
    p.x = x;
    p.y = y;
    p.step = ;
    Q.push(p);
    used[x][y] = ;
    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=; i<; i++)
        {
            q.x = p.x + dir[i][];
            q.y = p.y + dir[i][];
            if(q.x<m && q.x>= && q.y>= && q.y<n && used[q.x][q.y]!= && map[q.x][q.y] != '#')
            {
                used[q.x][q.y] = ;
                q.step = p.step + ;
                Q.push(q);
            }
        }
    }
}

int Prim(int cnt)
{
    int ans=;
    vis[] = ;
    for(int i=; i<=cnt; i++)
        dist[i] = D[][i];
    for(int i=; i<=cnt; i++)
    {
        int Min = INF, index = -;
        for(int j=; j<=cnt; j++)
            if(vis[j]== && Min > dist[j])
            {
                Min = dist[j];
                index = j;
            }
        if(index==-)break;
        ans += Min;
        vis[index] = ;
        for(int j=; j<=cnt; j++)
        {
            if(vis[j] ==  && dist[j] > D[index][j])
               dist[j] = D[index][j];
        }
    }
    return ans;
}

int main()
{
    int T, ans;
    scanf("%d", &T);
    while(T--)
    {
        memset(a, , sizeof(a));
        memset(vis, , sizeof(vis));

        memset(map, , sizeof(map));
        for(int i=; i<N; i++)
        {
            dist[i] = INF;
            for(int j=; j<N; j++)
                D[i][j] = INF;
            D[i][i] = ;
        }

        int cnt = ;
        scanf("%d%d ", &n,&m);
        for(int i=; i<m; i++)
        {
            gets(map[i]);
            for(int j=; j<n; j++)
            {
                if(map[i][j] <= 'Z' && map[i][j] >= 'A')
                    b[i][j] = cnt, cnt++;
            }
        }
        for(int i=; i<m; i++)
        {
            for(int j=; j<n; j++)
            {
                if(map[i][j] <= 'Z' && map[i][j] >= 'A')
                    bfs(b[i][j], i, j);
            }
        }
        ans = Prim(cnt-);
        printf("%d\n", ans);
    }
    return ;
}