链接:https://ac.nowcoder.com/acm/problem/22594
来源:牛客网
题号:NC22594
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 524288K,其他语言1048576K
64bit IO Format: %lld
题目描述
Island 发生了一场暴乱!现在 Rinne 要和 Setsuna 立马到地上世界去。
众所周知:Island 是有一些奇怪的城镇和道路构成的(题目需要,游戏党勿喷),有些城镇之间用双向道路连接起来了,且每条道路有它自己的距离。但是有一些城镇已经被派兵戒严,虽然主角可以逆天改命强闯,但是为了体验该游戏的平衡性,他们只能穿过不超过 K 次被戒严的城镇。
定义“穿过”:从一个戒严的点出发到达任意一个点,都会使得次数加1
现在他们想从 1 号城镇最快的走到 n 号城镇(即出口),现在他们想让你告诉他们最短需要走多少路。
输入描述:
第一行三个整数 n,m,k,分别表示城镇数量,边数量和最多能闯过被戒严的城市的次数。
接下来 n 行,每行一个整数 1 或 0,如果为 1 则表示该城市已被戒严,0 则表示相反。
接下来 m 行,每行三个数 u,v,w,表示在 u 城镇和 v 城镇之间有一条长度为 w 的双向道路。
输出描述:
输出一行一个数字,表示从 1 到 n 的最短路径,如果到达不了的话,请输出 -1。
解题思路
最短路求
1
−
n
1-n
1−n距离,每次求的时候维护一个
t
i
m
e
s
[
i
]
times[i]
times[i]用于记录从点
1
1
1出发到当前点最小长度的情况下最小的穿过戒严点次数
易错点
最后一个点(终点)不用穿过
输入样例
4 5 1
1
0
1
0
1 2 10
2 3 10
3 1 15
1 4 60
3 4 30
输出样例
60
AC代码
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define N 805
#define INF 0x3f3f3f3f
#define N2 100000
int edge[N][N],len[N],pre[N],times[N],vis[N],jie[N];
void solve(void){
int n,m,k,u,v,w;
cin >> n >> m >> k;
for(int i = 1;i <= n;i++) cin >> jie[i];
memset(edge,INF,sizeof(edge));
memset(len,INF,sizeof(len));
memset(times,INF,sizeof(times));
for(int i = 1;i <= n;i++) edge[i][i] = 0;
for(int i = 1;i <= m;i++){
cin >> u >> v >> w;
edge[u][v] = edge[v][u] = min(edge[u][v],w);
}
len[1] = 0,times[1] = jie[1];
int tmp,mn,now;
for(int i = 1;i <= n;i++){
tmp = -1,mn = len[0];
for(int j = 1;j <= n;j++){
if(!vis[j] && len[j] < mn && times[j] <= k) tmp = j,mn = len[j];
}
if(tmp == -1) break;
vis[tmp] = 1;
for(int j = 1;j <= n;j++){
//now 用于记录当前是否为最后一个点并且该点为戒严点(终点不用计算)
if(j == n && jie[j]) now = 1;
else now = 0;
if(len[j] > len[tmp] + edge[tmp][j] && times[tmp] + jie[j] - now <= k){
len[j] = len[tmp] + edge[tmp][j];
times[j] = times[tmp] + jie[j];
pre[j] = tmp;
}else if(len[j] == len[tmp] + edge[tmp][j] && times[tmp] + jie[j] - now <= k){
//到当前点最小长度相同的情况下最小穿过次数
times[j] = min(times[j],times[tmp] + jie[j]);
pre[j] = tmp;
}
}
}
if(len[n] >= len[0]) cout << -1;
else cout << len[n];
}
signed main(void){
//freopen("test.in","r",stdin);
ios::sync_with_stdio(false);
cin.tie(0);
int t = 1;
while(t--) solve();
return 0;
}