题目
给定一个 n 个点 m 条边的有向图,图中可能存在重边和自环, 边权可能为负数。
请你求出从 1 号点到 n 号点的最多经过 k 条边的最短距离,如果无法从 1 号点走到 n 号点,输出 impossible
。
注意:图中可能 存在负权回路 。
输入格式
第一行包含三个整数 n,m,k。
接下来 m 行,每行包含三个整数 x,y,z,表示存在一条从点 x 到点 y 的有向边,边长为 z。
点的编号为 1∼n。
输出格式
输出一个整数,表示从 1 号点到 n 号点的最多经过 k 条边的最短距离。
如果不存在满足条件的路径,则输出 impossible
。
数据范围
1≤n,k≤500
1≤m≤10000
1≤x,y≤n
任意边长的绝对值不超过 10000。
思路:
(十分暴力)将dist数组全部初始化为正无穷,其中dist[1] = 0;点1到它本身的距离为0。进行k次循环,每次循环使用backup对dist进行更新(相当于每次循环向前走一步),进行k次循环后,输出dist[n],如果此时的dist[n] >= 0x3f3f3f3f(表示点1到点n的距离为正无穷),则表示经过k步,不能从点1到达点n,输出impossible,否则输出从点1到点n的最短距离。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N = 1010;
const int M = 10010;
struct Edge{// 定义一个结构体,储存点a到点b的距离w
int a,b,w;
}edges[M];// 使用结构体数组进行储存
int n,m,k;// n 表示点数,m 表示边数,k表示走的步数
int dist[N],backup[N];// dist表示点i到点1的距离,backup表示的是dist的备份
int bellman_ford()
{
memset(dist,0x3f,sizeof(dist));//将点n到点1的距离改为无穷大
dist[1] = 0;// 点1到点1的距离为0
for(int i = 0; i < k; i ++)//
{
memcpy(backup,dist,sizeof(backup));// 将距离备份一下,使用备份的数据对dist进行更新,防止点进行串联
for(int j = 0; j < m; j ++)// 使用备份数据对dist进行更新
{
int a,b,w;
a = edges[j].a;
b = edges[j].b;
w = edges[j].w;
dist[b] = min(dist[b],backup[a] + w);// 更新数据时保留较小的距离
}
}
return dist[n];// 返回经过k步,点1到点n的最短距离
}
int main()
{
cin >> n >> m >> k;// 输入点数,边数,走的步数
for(int i = 0; i < m; i ++)
{
int a,b,w;// 点a到点b的距离w
cin >> a >> b >> w;// 输入点a到点b的距离w
edges[i].a = a;// 使用结构体数组储存起来
edges[i].b = b;
edges[i].w = w;
}
int t = bellman_ford();
if(t >= 0x3f3f3f3f / 2) cout << "impossible" << endl;// 如果1不能经过k步到达点n,n到1的距离为无穷大,输出“impossible”
else cout << t << endl;// 如果能到达则输出点1到点n最短距离。
}