TCZX--1501 搜索迷宫

题目描述：

在迷宫中，人们可以往北部，西部，东部和南部移动脚步。一个人从起点出发，希望迷宫里所有的人都能联系起来，并且联系的成本最低。注意：不需要两两联系，通过别人联系即可。
第一行包含两个整数 x，y 的行开始，使得 1≤x，y≤50。然后，跟随 y 行，每行 x 个字符。对于每个角色，空格“ ”代表开放空间，标记“＃”代表障碍墙，大写字母 “A” 代表人，大写字母 “S” 代表搜索的开始。迷宫的周长总是关闭的，也就是说，没有办法从 “S” 的坐标中走出来。迷宫中至多有 100 个人，每个人都可以到达。
输出描述：

输出一行包含成功搜索迷宫的最小成本

样例：

7 7
#####
#AAA###
#    A#
# S ###
#     #
#AAA###
#####

11

思路描述：

把各个点连接起来，构成树，再去跑最小生成树

用bfs跑出各个点之间的距离，再用prim或Kruskal跑出ans。 

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
#define debug(x) cout << #x << " = " << x << endl;
const int N = 2e5 + 5;
int a[101][101],step[101][101];
string tu[101];
int num, n, m;
int dx[4] = {0, 0, 1, -1};
int dy[4] = {1, -1, 0, 0};
int mod = 1e9 + 7;
void bfs(int i, int j){
    queue<pair<int, int>> q;
    q.push({i, j});
    vector<vector<int>>dist(101, vector<int>(101, 0));
    vector<vector<int>> st(101, vector<int>(101, 0));
    dist[i][j] = 0;
    st[i][j] = 1;
    while (!q.empty()){
        auto [x, y] = q.front();
        q.pop();
        if(a[x][y]){
            step[a[i][j]][a[x][y]] = dist[x][y];
        }
        for (int i = 0; i < 4; i++){
            int nx = x + dx[i];
            int ny = y + dy[i];
            if (nx >= 1 && nx <= n && ny >= 1 && ny <= m && tu[nx][ny] != '#' && !st[nx][ny]){
                st[nx][ny] = 1;
                q.push({nx, ny});
                dist[nx][ny] = dist[x][y] + 1;
            }
        }
    }
}
int prim(){
    vector<int> vis(101);
    vector<int>dis(101, 0x3f3f3f3f);
    dis[1] = 0;
    int ans = 0;
    
    for (int i = 1; i <= num;i++){
        int t = -1;
        for (int j = 1; j <= num; j++)
        {   
            if((t==-1||dis[j]<dis[t]) && !vis[j]){
                t = j;
            }
        }
        ans += dis[t];
        vis[t] = 1;
        for (int j = 1; j <= num; j++)
        {
            if (dis[j] > step[t][j])
            {
                dis[j] = step[t][j];
            }
        }
    }
    return ans;
}
void slove()
{
    cin >> n >> m;
    string s;
    getline(cin, s);//防吃

    for (int i = 1; i <= n; i++)
    {
        getline(cin, tu[i]);
        tu[i] = " " + tu[i];
    }
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++){
            cin >> tu[i][j];
            if( tu[i][j] == 'S'||tu[i][j]=='A'){
                a[i][j] = ++num;
            }
        }
    for (int i = 1; i <= n;i++){
        for (int j = 1; j <= m;j++){
            if(a[i][j]){
                bfs(i, j);
            }
        }
    }
    cout << prim();
}
signed main()
{
    ios::sync_with_stdio(0);
    std::cin.tie(0);
    std::cout.tie(0);
    // int n;
    // cin >> n;
    // while (n--)
    // {
    slove();
    // }
}

原题链接