洛谷 P4568 [JLOI2011]飞行路线(分层图+最短路变形)

本文链接：https://blog.csdn.net/weixin_45755679/article/details/117090776

在这里插入图片描述
初学分层图，写写模板题。

个人感觉分层图大部分会应用在最短路上，主要体现为题目会要求将图中的一些边的边权进行改变。比如选择 $k$ 条边不计算其贡献等等。这类题目的解题方法大部分都是在原先的 $d j$ 最短路算法上再增加几维状态，就像把这个图分为了好几层，每一层都由当前层或前一层的状态转移而来。

(以上纯属菜鸡理解，大佬勿喷)

题解：

因为要选择几条边不计贡献，那么我们就再增加一维状态， $d i s [i] [j]$ 表示达到点 $i$ 时，已经选择了 $j$ 条边不计入贡献的最短路。那么在跑最短路的时候就很好转移了，当前边要么不计入贡献， $d i s [v] [j] = d i s [u] [j] + w$ ,要么计入贡献 $d i s [v] [j + 1] = d i s [u] [j]$

代码：

#pragma GCC diagnostic error "-std=c++11"
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int> pii;
const int mod=1e9+7;
const int MAXN=2e5+5;
const int inf=0x3f3f3f3f;
int head[MAXN];
int vis[MAXN][15];
ll dis[MAXN][15];
int cnt;
struct edge
{
    int to;
    int w;
    int next;
}e[MAXN*2];
struct node
{
    int pos;
    ll val;
    int num;
    friend bool operator<(node a,node b)
    {
        return a.val>b.val;
    }
};
void add(int u,int v,int w)
{
    e[cnt].to=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt++;
}
void solve(int s,int k)
{
    memset(vis,0,sizeof vis);
    memset(dis,0x3f3f3f3f3f3f3f3f,sizeof dis);
    priority_queue<node> q;
    q.push(node{s,0,0});
    dis[s][0]=0; 
    while(!q.empty())
    {
        node now=q.top();
        q.pop();
        if(vis[now.pos][now.num]) continue;
        vis[now.pos][now.num]=1;
        for(int i=head[now.pos];i!=-1;i=e[i].next)
        {
            int v=e[i].to;
            if(dis[v][now.num]>now.val+e[i].w)
            {
                dis[v][now.num]=now.val+e[i].w;
                q.push({v,dis[v][now.num],now.num});
            }
            if(now.num<k&&dis[v][now.num+1]>now.val)
            {
                dis[v][now.num+1]=now.val;
                q.push({v,dis[v][now.num+1],now.num+1});
            }
        }
    }
}
int main()
{
    int n,m,k;
    cin>>n>>m>>k;
    int s,t;
    cin>>s>>t;
    memset(head,-1,sizeof head);
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        cin>>u>>v>>w;
        add(u,v,w);
        add(v,u,w);
    }
    solve(s,k);
    ll ans=1e18;
    for(int i=0;i<=k;i++)
    {
        ans=min(ans,dis[t][i]);
    }
    cout<<ans<<endl;
}