BFS+PRIM

转载请注明出处:優YoU http://user.qzone.qq.com/289065406/blog/1299324104

在一个y行 x列的迷宫中,有可行走的通路空格’ ‘,不可行走的墙’#’,还有两种英文字母A和S,现在从S出发,要求用最短的路径L连接所有字母,输出这条路径L的总长度。

一格的长度为1,而且移动的方法只有上、下、左、右,

所以在无任何墙的情况下(但“墙#”是必须考虑的,这里只是为了说明)

任意两个字母之间的距离就是直接把 横坐标之差 加上 纵坐标之差

注意的是,可行的路为 字母 和 空格

      不可行的路为 # 和 矩阵范围之外

根据题意的“分离”规则,重复走过的路不再计算

因此当使用prim算法求L的长度时,根据算法的特征恰好不用考虑这个问题(源点合并很好地解决了这个问题),L就是最少生成树的总权值W

由于使用prim算法求在最小生成树,因此无论哪个点做起点都是一样的,(通常选取第一个点),因此起点不是S也没有关系

所以所有的A和S都可以一视同仁,看成一模一样的顶点就可以了

最后要注意的就是 字符的输入

cin不读入空字符(包括 空格,换行等)

gets读入空格,但不读入换行符)

剩下的问题关键就是处理 任意两字母间的最短距离,由于存在了“墙#” ,这个距离不可能单纯地利用坐标加减去计算,必须额外考虑,推荐用BFS(广搜、宽搜),这是本题的唯一难点,因为prim根本直接套用就可以了

求 任意两字母间的最短距离 时不能直接用BFS求,

1、必须先把矩阵中每一个允许通行的格看做一个结点(就是在矩阵内所有非#的格都作为图M的一个顶点),对每一个结点i,分别用BFS求出它到其他所有结点的权值(包括其本身,为0),构造结点图M;

2、然后再加一个判断条件,从图M中抽取以字母为顶点的图,进而构造字母图N

这个判定条件就是当结点图M中的某点j为字母时,把i到j的权值再复制(不是抽离)出来,记录到字母图N的邻接矩阵中

3、剩下的就是对字母图N求最小生成树了

#include<stdio.h>
#include<iostream>
#include<cstdio>
#include<queue>
#include <vector>
#include<stack>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<climits>
#include<algorithm>
using namespace std;
#define ll unsinged long long
#define PI acos(-1.0)
#define ING 0x7fffffff
#define mod 1000000007
#define INF 3000

int dx[4]= {0,0,-1,1};
int dy[4]= {1,-1,0,0};
int node[105][105];
char ma[110][110];

int edge[110][110];
int num;
int n,m;
int que_x[1010],que_y[1010];
int vis[1010][1010];

void bfs(int i,int j)
{
    int head,rail;
    int dis[110][110];

    head=rail=0;
    memset(vis,0,sizeof(vis));
    memset(dis,0,sizeof(dis));

    que_x[rail]=i;
    que_y[rail]=j;
    rail++;
    vis[i][j]=1;

    while(head<rail)
    {
        int x=que_x[head];
        int y=que_y[head];
        head++;
        if(node[x][y])
            edge[node[i][j]][node[x][y]]=dis[x][y];

        for(int i=0; i<4; i++)
        {
            int mx=x+dx[i];
            int my=y+dy[i];
            if(mx<1||my<1||mx>m||my>n||vis[mx][my])
                continue;
            if(ma[mx][my]=='#')
                continue;
            vis[mx][my]=1;
            dis[mx][my]=dis[x][y]+1;
            que_x[rail]=mx;
            que_y[rail]=my;
            rail++;
        }
    }
}

void prim()
{
//    for(int i=1; i<=m; i++)
//    {
//        for(int j=1; j<=n; j++)
//        {
//            printf("%d ",edge[i][j]);
//        }
//        printf("\n");
//    }
    int mimi;
    int d[110];
    int v[110];
    memset(v,0,sizeof(v));
    for(int i=2; i<=num; i++)
    {
        d[i]=edge[1][i];
    }
    v[1]=1;
    d[1]=0;
    for(int i=1; i<=num-1; i++)
    {
        mimi=INF;
        int k=0;
        for(int j=1; j<=num; j++)
        {
            if(!v[j]&&d[j]<mimi)
            {
                k=j;
                mimi=d[j];
            }
        }
        v[k]=1;
        for(int j=1; j<=num; j++)
        {
            if(!v[j]&&edge[k][j]<d[j])
                d[j]=edge[k][j];
        }
    }
    int ans=0;
    for(int i=1; i<=num; i++)
    {
        ans+=d[i];
    }
    printf("%d\n",ans);
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        char s[51];
        gets(s);
        for(int i=1; i<=m; i++)
        {
            gets(ma[i]+1);
        }
        num=0;
        memset(node,0,sizeof(node));
        for(int i=1; i<=m; i++)
        {
            for(int j=1; j<=n; j++)
            {
                if(ma[i][j]=='A'||ma[i][j]=='S')
                {
                    node[i][j]=++num;
                }
            }
        }
//        for(int i=1; i<=m; i++)
//        {
//            for(int j=1; j<=n; j++)
//            {
//                printf("%c",ma[i][j]);
//            }
//            printf("\n");
//        }
        for(int i=1; i<=m; i++)
        {
            for(int j=1; j<=n; j++)
            {
                if(node[i][j])
                    bfs(i,j);
            }
        }
        prim();
    }
    return 0;
}

//0 6 6 2 0 0
//6 0 2 4 0 0
//6 2 0 4 0 0
//2 4 4 0 0 0
//0 0 0 0 0 0
//
//0 1 2 5 3 4 5
//1 0 1 4 2 5 4
//2 1 0 3 3 6 5
//5 4 3 0 4 7 6
//3 2 3 4 0 3 2
//4 5 6 7 3 0 1
//5 4 5 6 2 1 0
posted @ 2016-03-24 23:03  see_you_later  阅读(120)  评论(0编辑  收藏  举报