POJ-3026(图上的最小生成树+prim算法+gets函数使用)

Borg Maze

POJ-3026

• 一开始看到这题是没有思路的，看了题解才知道和最小生成树有关系。
• 题目的意思是每次走到一个A或者S就可以分为多个部分继续进行搜索。这里就可以看出是从该点分出去的不同路径。
• 所以首先需要使用bfs求出每对顶点的最短路来，但是在这个过程中，我出了个bug，导致调试半天，就是那里bfs上下左右走的时候的x,y都用的原来的。
• 最小生成树就是使用prim算法，该算法和dijikstra算法特别像，唯一有区别的就是mincost函数的定义。

#include<iostream>
#include<cmath>
#include<cstring>
#include<string>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int INF=0X3F3F3F3F;
int m,n;
char map[55][55];
int cnt[55][55];
bool vis[55][55];
int ma[202][202];
int dir[4][2]={{-1,0},{0,-1},{1,0},{0,1}};
int total;
struct node{
    int x;
    int y;
    int dis;
};
bool in(int x,int y){
    return x>=1&&x<=n&&y>=1&&y<=m;
}
void bfs(int x,int y){
    memset(vis,0,sizeof(vis));
    queue<node> q;
    q.push(node{x,y,0});
    vis[x][y]=1;
    while(!q.empty()){
        node temp=q.front();
        q.pop();
        if(cnt[temp.x][temp.y]>0){//该结点是A或者S
            ma[cnt[x][y]][cnt[temp.x][temp.y]]=temp.dis;
        }
        for(int i=0;i<4;i++){
            int tx=dir[i][0]+temp.x;
            int ty=dir[i][1]+temp.y;
           // cout<<tx<<" "<<ty<<endl;
            if(in(tx,ty)&&!vis[tx][ty]&&map[tx][ty]!='#'){
                //cout<<tx<<" "<<ty<<" "<<map[tx][ty]<<endl;
                vis[tx][ty]=1;
                q.push(node{tx,ty,temp.dis+1});
                //cout<<tx<<" "<<ty<<endl;
            }
        }
    }
    //cout<<endl;
}
bool viss[202];
int mincost[202];
int prim(int s){
    memset(viss,0,sizeof(viss));
    memset(mincost,INF,sizeof(mincost));
    mincost[s]=0;
    int ans=0;
    for(int i=1;i<=total;i++){
        int v=-1;
        int mins=INF;
        for(int j=1;j<=total;j++){
            if(!viss[j]&&mincost[j]<mins){
                mins=mincost[j];
                v=j;
            }
        }
        if(v==-1)
            break;
        viss[v]=1;
        ans+=mins;//mincost[v]
        for(int j=1;j<=total;j++){
            mincost[j]=min(mincost[j],ma[v][j]);
        }
    }
    return ans;
}
int main(){
    int t;
    cin>>t;
    while(t--){
        cin>>m>>n;
        total=0;
        memset(cnt,0,sizeof(cnt));
        gets(map[0]);
        for(int i=1;i<=n;i++){
            gets(map[i]+1);//这里从第i行第2个字符开始读入
        }
        for(int i=1;i<=n;i++){
            //cout<<map[i]<<endl;
            for(int j=1;j<=m;j++){
                if(map[i][j]=='S'||map[i][j]=='A'){
                    cnt[i][j]=++total;
                    //cout<<map[i][j]<<" ";
                }
                //cout<<map[i][j]<<" ";
            }
            //cout<<endl;
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(cnt[i][j]>0){
                    bfs(i,j);
                }
            }
        }
        // for(int i=1;i<=total;i++){
        //     for(int j=1;j<=total;j++){
        //         cout<<ma[i][j]<<" ";
        //     }
        //     cout<<endl;
        // }
        cout<<prim(1)<<endl;
    }
    return 0;
}