码迷,mamicode.com
首页 > 其他好文 > 详细

【模板】倍增+Floyd

时间:2018-11-20 00:09:50      阅读:157      评论:0      收藏:0      [点我收藏+]

标签:include   opera   print   floyd   mem   \n   and   通过   end   

题目大意:给定一个 N 个顶点的邻接矩阵、起点顶点、终点顶点,求至少经过 K 条边(边可以重复)从起点到终点的最短路长度,若不能到达,输出 -1。

题解:至少经过 K 条边和恰好经过 K 条边的初始条件不同,因为至少经过 1 条边的任意两点最短路就是通过 Floyd 算法算出的矩阵,而恰好经过 K 条边的任意两点的最短路则是最初的邻接矩阵。不过两个算法的矩阵幂算法是相同的,可以用快速幂加速递推。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=101;
const int inf=0x3f3f3f3f;

int n,k,st,ed;
struct matrix{
    int data[maxn][maxn];
    matrix(){memset(data,0x3f,sizeof(data));}
    inline int* operator[](int i){return data[i];}
    friend matrix operator*(matrix& x,matrix& y){
        matrix z;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                for(int k=1;k<=n;k++)
                    z[i][j]=min(z[i][j],x[i][k]+y[k][j]);
        return z;
    }
}d,ans;

void read_and_parse(){
    scanf("%d%d%d%d",&n,&k,&st,&ed);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++){
            scanf("%d",&d[i][j]);
            if(d[i][j]==-1)d[i][j]=inf;
        }
}

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

void solve(){
    floyd();
    ans=d,k--; 
    for(;k;k>>=1,d=d*d)if(k&1)ans=ans*d;
    if(ans[st][ed]==inf)puts("-1");
    else printf("%d\n",ans[st][ed]);
} 

int main(){
    read_and_parse();
    solve();
    return 0;
}

【模板】倍增+Floyd

标签:include   opera   print   floyd   mem   \n   and   通过   end   

原文地址:https://www.cnblogs.com/wzj-xhjbk/p/9986334.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!