bellman-ford AcWing 853. 有边数限制的最短路

bellman-ford AcWing 853. 有边数限制的最短路

原题链接

AcWing 853. 有边数限制的最短路

算法标签

最短路 Bellman-Ford

思路

由题意 存在负边， 因此不能采用Dijkstra
由于具有边数限制
故采用Bellman-Ford
Bellman-Ford算法

对于存在负权回路的路径
可能不存在最短路
例如下图
1至5最短路 不存在 由于每次在2, 3, 4回路中循环一次， 路几个长度减1

代码

#include<bits/stdc++.h>
#define int long long
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>b;--i)
using namespace std;
const int N = 10005;
// p存储节点祖宗节点 d存储当前节点据祖宗节点的距离
int lst[N], dis[N];
int n,m,k;
inline int read(){
   int s=0,w=1;
   char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
   while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
   return s*w;
}
void put(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>=10) put(x/10);
    putchar(x%10^48);
}
struct Edge{
    int a, b, c;
}edge[N];
int be(){
    memset(dis, 0x3f3f, sizeof dis);
    dis[1]=0;
    rep(i, 0, k){
        memcpy(lst, dis, sizeof dis);
        rep(j, 0, m){
            Edge e=edge[j];
            dis[e.b]=min(dis[e.b], lst[e.a]+e.c);
        }
    }
    if(dis[n]>0x3f3f3f3f/2){
        return -1;
    }else{
        return dis[n];
    }
}
signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    n=read(), m=read(), k=read();
    rep(i, 0, m){
        int x=read(), y=read(), z=read();
        edge[i]={x, y, z};
    }
    if(be()==-1){
        puts("impossible");
    }else{
        printf("%lld", dis[n]);
    }
    return 0;
}

原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈