zoukankan      html  css  js  c++  java
  • 【模板】倍增+Floyd

    题目大意:给定一个 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
    ",ans[st][ed]);
    } 
    
    int main(){
    	read_and_parse();
    	solve();
    	return 0;
    }
    
  • 相关阅读:
    js-数组的原型拓展
    java-单例模式的java连接池
    java-基本数据类型值范围
    rabbitmq
    从哪跌倒从哪爬起,千里之行始于足下
    create python project steps
    常用代码片段
    shell脚本常用(记)
    shell学习
    maven工具使用
  • 原文地址:https://www.cnblogs.com/wzj-xhjbk/p/9986334.html
Copyright © 2011-2022 走看看