Rinne Loves Graph

链接：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$距离，每次求的时候维护一个$times[i]$用于记录从点$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;
}