poj 3026 （bfs+prim）

这个题有一个意思描述使的题目简单了。。囧。。开始还以为数据水呢。其实是我题目没看清楚。。。只能在A,S的时候才能分成很多的路径。

1
5 8
A###A
 ###
     
## ##
## ##
## ##
##S##
#####

比如这个数据输出的是16。ps:假如没有那个条件的话，就是应该输出12。。。。(怎么做？。。)

这题的代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>

using namespace std;

#define inf 0x7ffffff
#define MAXN 52
int map[MAXN<<1][MAXN<<1];
char c[MAXN][MAXN];
int id[MAXN][MAXN];
int y,x,cnt;
int dir[4][2]={0,-1,0,1,-1,0,1,0};

struct point
{
    int x,y;
    int step;
    point(){}
    point(int a,int b,int c){x=a,y=b,step=c;}
};

bool is_ok(int i,int j)
{
    if(i>=0 && i<x && j>=0 && j<y)
        return true;
    return false;
}

void bfs(int tx,int ty)
{
    queue<point> q;
    while(!q.empty()) q.pop();
    bool vis[MAXN][MAXN];
    point sp=point(tx,ty,0);
    memset(vis,0,sizeof(vis));
    vis[tx][ty]=1;
    q.push(sp);
    while(!q.empty())
    {
        point cur,next;
        cur=q.front();q.pop();
        if(c[cur.x][cur.y]=='A' || c[cur.x][cur.y]=='S')
            map[id[tx][ty]][id[cur.x][cur.y]]=cur.step;
        for(int i=0;i<4;i++)
        {
            next.x=cur.x+dir[i][0];
            next.y=cur.y+dir[i][1];
            next.step=cur.step+1;
            if(!is_ok(next.x,next.y)) continue;
            if(c[next.x][next.y]=='#') continue;
            if(vis[next.x][next.y]) continue;
            vis[next.x][next.y]=1;
            q.push(next);
        }
    }
}
//---------------prim--------------
int prim(int N)
{
    int i,j;
    int ans=0;
    int dis[MAXN<<1];
    bool vis[MAXN<<1];
    for(i=0;i<N;i++)
    {
        vis[i]=0;
        dis[i]=map[0][i];
    }
    vis[0]=1;
    for(i=0;i<N-1;i++)
    {
        int MIN=inf,flag=-1;
        for(j=0;j<N;j++)
        {
            if(!vis[j] && dis[j]<MIN)
            {
                flag=j;
                MIN=dis[j];
            }
        }
        vis[flag]=1;
        ans+=MIN;//答案
        for(j=0;j<N;j++)
            if(!vis[j] && map[flag][j]<dis[j])
                dis[j]=map[flag][j];
    }
    return ans;
}
//---------------------------------


int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        cnt=0;
        scanf("%d%d",&y,&x);
        gets(c[0]);
        memset(c,0,sizeof(c));
        for(int i=0;i<x;i++)
        {
            gets(c[i]);
            for(int j=0;j<y;j++)
                if(c[i][j]=='A' || c[i][j]=='S')
                    id[i][j]=cnt++;
        }
        for(int i=0;i<x;i++)
        {
            for(int j=0;j<y;j++)
                if(c[i][j]=='A' || c[i][j]=='S')
                    bfs(i,j);
        }
        /*------------------
        for(int i=0;i<cnt;i++)
        {
            for(int j=0;j<cnt;j++)
                cout<<map[i][j];
            cout<<endl;
        }
        */
        printf("%d\n",prim(cnt));

    }
}
/*
1
5 8
A###A
 ###
     
## ##
## ##
## ##
##S##
#####
 */