PIPIOJ 1183: 信使PIPI 多次bfs+最小生成树

最新推荐文章于 2022-02-18 22:34:05 发布

化身孤岛的鲸o

最新推荐文章于 2022-02-18 22:34:05 发布

阅读量285

点赞数 2

CC 4.0 BY-SA版权

分类专栏： PIPIOJ 最小生成树 BFS

本文链接：https://blog.csdn.net/weixin_44123362/article/details/109063618

PIPIOJ 同时被 3 个专栏收录

43 篇文章

订阅专栏

最小生成树

11 篇文章

订阅专栏

BFS

10 篇文章

订阅专栏

该博客主要介绍了一种解决图论问题的方法，通过将字符矩阵转换为邻接矩阵，并利用Prim算法求解最小生成树。代码实现中包含了哈希映射、BFS遍历和邻接矩阵的构建过程，适用于理解图算法和数据结构。

题目：

http://39.106.164.46/problem.php?id=1183

思路：

（1）将每个为’C’或者小写’c’的点哈希成1~cnt的序号，代表图中有cnt个点。
（2）多次bfs将所给的矩阵化为邻接矩阵。
（3）然后用prim算法求最小生成树。

代码如下：

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<cmath>
#include<queue>
#include<cstring>
#include<vector>
#include<stack>
#include<map>
#include<set>
#define MAX 105
#define INF 0x3f3f3f3f
typedef long long ll;
using namespace std;

int n,id=0;
char ch[MAX][MAX];
int vis[MAX][MAX],dir[4][2]={{0,-1},{0,1},{-1,0},{1,0}};
int to[MAX][MAX],mp[MAX][MAX],dis[MAX],visit[MAX];

struct node{
    int x,y,t;
};

bool judge(int x,int y){
    if(x<0||x>n-1||y<0||y>n-1) return false;
    return true;
}

void bfs(int x,int y){
    memset(vis,0,sizeof(vis));
    queue<node> q;
    node tmp;
    tmp.x=x,tmp.y=y,tmp.t=0;
    q.push(tmp);
    vis[x][y]=1;
    while(!q.empty()){
        node top=q.front();
        q.pop();
        for(int i=0;i<4;i++){
            int xx=top.x+dir[i][0],yy=top.y+dir[i][1];
            tmp.x=xx,tmp.y=yy,tmp.t=top.t+1;
            if(judge(xx,yy)&&ch[xx][yy]!='#'&&vis[xx][yy]==0){
                if(ch[xx][yy]=='c'||ch[xx][yy]=='C'){
                    mp[to[x][y]][to[xx][yy]]=tmp.t;
                }
                vis[xx][yy]=1;
                q.push(tmp);
            }
        }
    }
}

int prim(){
    int ans=0;
    memset(dis,INF,sizeof(dis));
    memset(visit,0,sizeof(visit));
    visit[1]=1;
    for(int i=1;i<=id;i++){
        dis[i]=mp[1][i];
    }
    for(int i=0;i<id-1;i++){
        int u=-1,minl=INF;
        for(int j=1;j<=id;j++){
            if(visit[j]==0&&dis[j]<minl){
                minl=dis[j];
                u=j;
            }
        }
        if(u==-1) return -1;
        visit[u]=1;
        ans+=dis[u];
        for(int v=1;v<=id;v++){
            if(visit[v]==0&&mp[u][v]<dis[v]){
                dis[v]=mp[u][v];
            }
        }
    }
    return ans;
}

int main(){
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%s",ch[i]);
    }
    int s,t;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(ch[i][j]=='C'){
                s=i,t=j;
                to[i][j]=++id;
            }
            if(ch[i][j]=='c'){
                to[i][j]=++id;
            }
        }
    }
    for(int i=1;i<=id;i++){
        for(int j=1;j<=id;j++){
            if(i==j) mp[i][j]=0;
            else mp[i][j]=INF;
        }
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(ch[i][j]=='C'||ch[i][j]=='c'){
                bfs(i,j);
            }
        }
    }
    int res=prim();
    cout<<res;
    return 0;
}