牛站（恰好经过k条边的最短路）

最新推荐文章于 2025-04-14 16:15:18 发布

原创最新推荐文章于 2025-04-14 16:15:18 发布 · 694 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#图论 #算法

图论专栏收录该内容

7 篇文章

订阅专栏

这篇博客介绍了一种利用快速幂算法优化Floyd-Warshall算法来求解最短路径的问题。通过将每一步的最短路径矩阵相乘，可以将计算复杂度降低到对数级别。在给定边的权重和起点、终点后，程序能够快速计算出经过指定步数的最短路径。

牛站

类似于矩阵快速幂，在floyd中，d[i][j] = min(d[i][j],d[i][k] + d[k][j])表示将d[i][j]更新成可经过k个点的最短路，在这题中，d[i][j]表示经过k条边的最短路。所以对于给定的k，利用快速幂将k将为logk，对每一位1更新res。（记得离散化）。具体看注释。

#include<bits/stdc++.h>
using namespace std;

const int N = 210;

int k,m,s,e,n;
int g[N][N],res[N][N];

void mul(int c[][N],int a[][N],int b[][N])
{
    int t[N][N];
    memset(t,0x3f,sizeof(t));

    for(int k = 1; k <= n; k ++ )
        for(int i = 1; i <= n; i ++ )
            for(int j = 1; j <= n; j ++ )
                t[i][j] = min(t[i][j],a[i][k] + b[k][j]);
                //这里假设题目给的k为5，a代表的res表示走一步，b代表的g走四步
                //所以t表示i到k走一步，k到j走四步，即i到j走五步的最短距离，完成更新

    memcpy(c,t,sizeof(t));
}

void qsm()
{
    memset(res,0x3f,sizeof(res));
    for(int i = 1; i <= n; i ++ )
        res[i][i] = 0;//res是单位元，假设k为1，操作一次mul，那么res变为g

    while(k)
    {
        if(k & 1)
            mul(res,res,g);
        mul(g,g,g);//g表示i到j走k步的最短距离
        k >>= 1;
    }
}

int main()
{
    cin>>k>>m>>s>>e;
    memset(g,0x3f,sizeof(g));//为什么对角线不为0，因为i到i如果为零，
                             //那么矩阵乘法那里的t[i][i]永远不会更新,
                             //但i到i走k步不一定是0
    map<int,int>mp;
    if(!mp.count(s))
        mp[s] = ++ n;
    if(!mp.count(e))
        mp[e] = ++ n;
    s = mp[s];
    e = mp[e];

    while(m -- )
    {
        int a,b,c;
        cin>>c>>a>>b;
        if(!mp.count(a))
            mp[a] = ++ n;
        if(!mp.count(b))
            mp[b] = ++ n;
        a = mp[a];
        b = mp[b];
        g[a][b] = g[b][a] = min(g[a][b],c);
    }

    qsm();

    cout<<res[s][e];
    return 0;
}