分层图最短路 P4568 [JLOI2011]飞行路线

本人也是初学最短路，欢迎交流。

关于dijkstra模板

P4568 [JLOI2011]飞行路线

原题链接：传送门

思路：

1. 常规求最短路，不过这次可以选择不超过K条边，使这些边的权值变成0；
2. 暑假多校遇到的题目，当时还是一个只会贴膜的少年，现在理解了一番最短路，加上分层图的扩展，也算是能解决这道题目了。
3. 分层图思想，在原本的二维图上面，再加一维，建(k+1)*n个点，部分操作会在代码中注释，层与层之间的边权为0对应把边权变成0的操作，最后要把每一层的终点连接起来其边权为0，防止选择小于k条边就到达终点了。
   
   代码如下：

#include<bits/stdc++.h>
using namespace std;
const int manx=2e6+5+100000; //注意都开大一点 因为要多建很多边 各种数组都受到了影响
const int mamx=2e6+5+500000;
const int maxd=2e6+5+100000;
int n,m,s,e,k,indexs=0;
int head[manx],d[maxd];
bool vis[manx];
priority_queue<pair<int,int> >q;
struct node{
    int v,w,next;
}a[mamx];
void add(int u,int v, int w)
{
    a[++indexs].v=v;
    a[indexs].w=w;
    a[indexs].next=head[u];
    head[u]=indexs;
}
void dij() //常规dij
{
    memset(d,0x3f,sizeof(d));
    memset(vis,0,sizeof(vis));
    d[s]=0;
    q.push(make_pair(0,s));
    while(q.size())
    {
        int u=q.top().second;
        q.pop();
        if(vis[u]) continue;
        vis[u]=1;
        for(int i=head[u];i;i=a[i].next)
        {
            int v=a[i].v,w=a[i].w;
            if(d[v]>d[u]+w)
                d[v]=d[u]+w,q.push(make_pair(-d[v],v));
        }
    }
}
int main()
{
    scanf("%d%d%d%d%d",&n,&m,&k,&s,&e);
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
        add(v,u,w); //无向图
        for(int j=1;j<=k;j++)
        {
            add(u+j*n,v+j*n,w);	  //不操作的时候进行原图扩展
            add(v+j*n,u+j*n,w);   //不操作的时候进行原图扩展
            add(v+j*n-n,u+j*n,0); //层与层之间 这个0代表 k的操作
            add(u+j*n-n,v+j*n,0); //层与层之间 这个0代表 k的操作
        }
    }
    for(int i=1;i<=k;i++) add(e+(i-1)*n,e+i*n,0); //终点连接起来
    dij();
    cout<<d[e+k*n];
    return 0;
}