AcWing 5268.最短距离

最新推荐文章于 2024-10-11 12:01:25 发布

akeyy

最新推荐文章于 2024-10-11 12:01:25 发布

阅读量65

点赞数

分类专栏： AcWing算法题解文章标签：算法图论

本文链接：https://blog.csdn.net/akeyy/article/details/134772806

版权

AcWing算法题解专栏收录该内容

9 篇文章 0 订阅

订阅专栏

题目描述

原题链接：5268.最短距离

解题思路

存储方式分析

如果要是两个点的距离为0，那么就说明这两个点属于同一类。假如将所有距离为0的点合并为一个集合，那么就可以用并查集处理。
ac5268_0
如图，1和2的距离为0，但是不属于同一类，此时就可以将这两类合并。这是因为，1和3为同一类，3到1的距离为0，那么3到2的距离就也为0。同理，2类内的每一个点到1类内的每一个点的距离也为0。

时间复杂度分析

由于 $\le n \le10^5, 0 \le m \le 10^5 ,1 \le k \le 500$ ，因此 $o(n)+o(k^3)$ 可以通过。考虑在合并距离为0的点之后利用一次Floyd算法，计算出类与类之间的最短距离。

算法流程

ac5268_1

1.初始化类

for(int i = 1, j = 0; i <= ks; i++)
{
    int c;
    cin >> c;
    while(c--)
    {
        id[++j] = i;
    }
}

2.合并距离为0的点

memset(d, 0x3f, sizeof d);
for(int i = 1; i <= ks; i++) d[i][i] = 0;
for(int i = 1; i <= m; i++)
{
    int l, r, s;
    cin >> l >> r >> s;
    if(s == 0) unionset(l, r);
    int x = id[l], y = id[r];
    d[x][y] = d[y][x] = min(d[x][y], s);
}

3.判断类内距离是否为0

bool check()
{
    for(int i = 2; i <= n; i++)
    {
        if(id[i] == id[i-1] && find(i) != find(i-1)) return false;
    }
    return true;
}

4.Floyd算法计算类间距离

for(int k = 1; k <= ks; k++)
{
    for(int i = 1; i <= ks; i++)
    {
        for(int j = 1; j <= ks; j++)
            d[i][j] = min(d[i][k] + d[k][j], d[i][j]);
    }
}

c++代码

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 100100;
int n, m, ks, fa[N], id[N], d[501][501], INF = 0x3f3f3f3f;
int find(int x)
{
    if(fa[x] == x) return x;
    else return fa[x] = find(fa[x]);
}
void unionset(int x, int y)
{
    x = find(x), y = find(y);
    fa[x] = fa[y];
}
bool check()
{
    for(int i = 2; i <= n; i++)
    {
        if(id[i] == id[i-1] && find(i) != find(i-1)) return false;
    }
    return true;
}
int main()
{
    cin >> n >> m >> ks;
    for(int i = 1; i <= n; i++) fa[i] = i;
    for(int i = 1, j = 0; i <= ks; i++)
    {
        int c;
        cin >> c;
        while(c--)
        {
            id[++j] = i;
        }
    }
    memset(d, 0x3f, sizeof d);
    for(int i = 1; i <= ks; i++) d[i][i] = 0;
    for(int i = 1; i <= m; i++)
    {
        int l, r, s;
        cin >> l >> r >> s;
        if(s == 0) unionset(l, r);
        int x = id[l], y = id[r];
        d[x][y] = d[y][x] = min(d[x][y], s);
    }
    if(!check()) puts("No");
    else 
    {
        puts("Yes");
        for(int k = 1; k <= ks; k++)
        {
            for(int i = 1; i <= ks; i++)
            {
                for(int j = 1; j <= ks; j++)
                    d[i][j] = min(d[i][k] + d[k][j], d[i][j]);
            }
        }
        for(int i = 1; i <= ks; i++)
        {
            for(int j = 1; j <= ks; j++)
            {
                if(d[i][j] == INF) d[i][j] = -1;
                cout << d[i][j] << ' ';
            }
            puts("");
        }
    }
    
}