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

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

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

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

题解:

因为要选择几条边不计贡献,那么我们就再增加一维状态, d i s [ i ] [ j ] dis[i][j] dis[i][j]表示达到点 i i i 时,已经选择了 j j j 条边不计入贡献的最短路。那么在跑最短路的时候就很好转移了,当前边要么不计入贡献, d i s [ v ] [ j ] = d i s [ u ] [ j ] + w dis[v][j]=dis[u][j]+w dis[v][j]=dis[u][j]+w ,要么计入贡献 d i s [ v ] [ j + 1 ] = d i s [ u ] [ j ] dis[v][j+1]=dis[u][j] dis[v][j+1]=dis[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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值