Borg Maze-POJ3026(bfs+最小生成树)

http://poj.org/problem?id=3026

如果一个一个普通搜处理不好的话会超时  可以连到一块搜

我觉得这个方法特别好

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<iostream>
#include<queue>
#include<algorithm>
#define N 105
#define INF 0x3f3f3f3f
using namespace std;
int dis[][]={{,},{-,},{,},{,-}};
int G[N][N],d[N],vis[N][N],m,n,v[N],b[N][N];
char F[N][N];
struct node
{
    int x,y,temp;
}a;

void Inn()
{
    int i,j;
    for(i=;i<=N;i++)
    {
        d[i]=INF;
        for(j=;j<=N;j++)
            G[i][j]=INF;
    }
    memset(b,,sizeof(b));
    memset(F,,sizeof(F));
}

void bfs(int s,int x,int y)
{
    int i;
    memset(vis,,sizeof(vis));
    queue<node>Q;
    a.x=x;
    a.y=y;
    a.temp=;
    Q.push(a);
    vis[a.x][a.y]=;
    while(Q.size())
    {
        node q,p;
        q=Q.front();
        Q.pop();
        if(F[q.x][q.y]=='A' || F[q.x][q.y]=='S')
            G[s][b[q.x][q.y]]=q.temp;
        for(i=;i<;i++)
        {
            p.x=q.x+dis[i][];
            p.y=q.y+dis[i][];

            if(F[p.x][p.y]!='#' && p.x<m&&p.y<n&&p.x>=&&p.y>=&&!vis[p.x][p.y])
            {
                p.temp=q.temp+;
                Q.push(p);
                vis[p.x][p.y]=;
            }
        }
    }
}

int prim(int k)
{
    memset(v,,sizeof(v));
    int i,j,ans=;
    for(i=;i<=k;i++)
    {
        d[i]=G[][i];
    }
    v[]=;
    for(i=;i<k;i++)
    {
        int Min=INF;
        int dist;
        for(j=;j<=k;j++)
        {
            if(!v[j] && d[j]<Min)
            {
                Min=d[j];
                dist=j;
            }
        }
        v[dist]=;
        ans+=Min;
        for(j=;j<=k;j++)
        {
            if(!v[j] && d[j]>G[dist][j])
                d[j]=G[dist][j];
        }
    }
    return ans;
}

int main()
{
    int T,i,j;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d %d ",&n,&m);
        Inn();
        int k=;
        for(i=;i<m;i++)
        {
            for(j=;j<n;j++)
            {
                scanf("%c",&F[i][j]);
                if(F[i][j]=='A' || F[i][j]=='S')
                {
                    b[i][j]=k;
                    k++;
                }
            }
            getchar();
        }
        for(i=;i<m;i++)
        {
            for(j=;j<n;j++)
            {
                if(F[i][j]=='A' || F[i][j]=='S')
                    bfs(b[i][j],i,j);
            }
        }
        printf("%d\n",prim(k-));
    }
    return ;
}