NC22594 Rinne Loves Graph （拆点+最短路）

题目描述
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
输入

4 5 1
1
0
1
0
1 2 10
2 3 10
3 1 15
1 4 60
3 4 30

输出

60

思路：
将每个点拆成经过pre个点
跑最短路就行了
代码：

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
typedef pair<int,int>PII;
const int N=815,M=8005;
int dist[N][15];    // dist表示每个点到起点的距离, q 是队列
int h[N], e[M], w[M],ne[M], idx;       // 邻接表
bool st[N][15];     // 存储每个点是否在队列中
bool arr[N];
int n,m,k;
void add(int a, int b, int c)
{
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
}
int dijkstra()
{
    memset(dist,0x3f,sizeof dist);
    priority_queue<PII,vector<PII>,greater<PII>>q;
    dist[1][0]=0;
    q.push({0,1+0*n});
    while(!q.empty())
    {
        PII t=q.top();
        q.pop();
        int pos=t.second%n;
        int fre=t.second/n;
        int dis=t.first;
        if(st[pos][fre])
        {
            continue;
        }
        st[pos][fre]=true;
        for(int i=h[pos];i!=-1;i=ne[i])
        {
            int j=e[i];
            int ver=0;
            if(arr[pos])
            {
                ver=1;
            }
            if(ver+fre>k)
            {
                continue;
            }
            if(dist[j][ver+fre]>dis+w[i])
            {
                dist[j][ver+fre]=dis+w[i];
                q.push({dist[j][ver+fre],j+(ver+fre)*n});
            }
        }
    }
    int res=0x3f3f3f3f;
    for(int i=0;i<=k;i++)
    {
        res=min(dist[n][i],res);
    }
    if(res==0x3f3f3f3f)
    {
        return -1;
    }
    return res;
}
int main()
{
    memset(h,-1,sizeof h);
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>arr[i];
    }
    for(int i=0;i<m;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
        add(b,a,c);
    }
    cout<<dijkstra();
}