poj 3026 Borg Maze bfs建图+最小生成树

题目说从S开始，在S或者A的地方可以分裂前进。 想一想后发现就是求一颗最小生成树。

首先bfs预处理得到每两点之间的距离，我的程序用map做了一个映射，将每个点的坐标映射到1-n上，这样建图比较方便。

然后一遍prime就够了。注意用gets（）读入地图的时候，上面还要用一个gets（）接住无用的空格。。（为啥不用getchar？0 0，看了讨论版才知道，

因为空格很多………………）

 

#include<cstdio>

#include<algorithm>

#include<cstring>

#include<map>

#include<iostream>

using namespace std;

#define INF 0x3f3f3f3f

char map1[55][55];

int head,tail;

int x,y;

map<int,int> M;

struct node

{

    int x,y;

    int dis;

}q[100000];

int top;

int start;

int dx[]={-1,1,0,0};

int dy[]={0,0,-1,1};

int dis[150][150];

bool vis[150][150];

void bfs(int k,int st)

{

    memset(vis,0,sizeof(vis));

    head=tail=0;

    node tmp,tt;

    tmp.x=st/x;

    tmp.y=st%x;

    vis[tmp.x][tmp.y]=1;

    tmp.dis=0;

    int tot=0;

    q[tail++]=tmp;

    int f;

    while(head<tail)

    {

        tmp=q[head];

        for(int d=0;d<4;d++)

        {

            tt=tmp;

            tt.x+=dx[d];

            tt.y+=dy[d];

            tt.dis++;

            if(map1[tt.x][tt.y]!='#'&&!vis[tt.x][tt.y])

            {

                vis[tt.x][tt.y]=1;

                if(f=M[tt.x*x+tt.y])

                {

                    tot++;

                    dis[k][f]=tt.dis;

                }

                q[tail++]=tt;

                if(tot==top) return;

            }

        }

        head++;

    }

}

int n,sum;

int visit[150],d[150];

void prime(int n)

{

	int i,j,min,v;

	sum=0;

	for(i=1;i<=n;i++)

	{

		d[i]=dis[1][i];

		visit[i]=0;

	}

	visit[1]=1;

	for(i=1;i<n;i++)

	{

		min=INF;

		v=1;

		for(j=1;j<=n;j++)

		{

			if(!visit[j]&&min>d[j])

			{

				min=d[j];

				v=j;

			}

		}

		sum+=min;

		visit[v]=1;

		for(j=1;j<=n;j++)

		{

			if(!visit[j]&&dis[v][j]<d[j])

				d[j]=dis[v][j];

		}

	}

}

int main()

{

    char tmp[1000];

    int cas;

    scanf("%d",&cas);

    while(cas--)

    {

        memset(dis,0x3f,sizeof(dis));

        M.clear();

        top=0;

        scanf("%d%d",&x,&y); gets(tmp);//接住空格

        for(int i=0;i<y;i++){

            gets(map1[i]);

        }

        for(int i=1;i<y-1;i++)

        {

            for(int j=1;map1[i][j];j++)

            {

                if(map1[i][j]=='A'||map1[i][j]=='S')

                {

                    top++;

                    if(map1[i][j]=='S') start=top;

                    M[i*x+j]=top;

                    dis[top][top]=0;



                }

            }

        }

        for(int i=1;i<y-1;i++)

        {

            for(int j=1;map1[i][j];j++)

            {

                if(map1[i][j]=='A'||map1[i][j]=='S')

                {

                    bfs(M[i*x+j],i*x+j);

                }

            }

        }

        prime(top);

        printf("%d\n",sum);

    }

    return 0;

}