P4568 [JLOI2011]飞行路线

题目描述

Alice和Bob现在要乘飞机旅行，他们选择了一家相对便宜的航空公司。该航空公司一共在nn个城市设有业务，设这些城市分别标记为00到n-1n−1，一共有mm种航线，每种航线连接两个城市，并且航线有一定的价格。

Alice和Bob现在要从一个城市沿着航线到达另一个城市，途中可以进行转机。航空公司对他们这次旅行也推出优惠，他们可以免费在最多kk种航线上搭乘飞机。那么Alice和Bob这次出行最少花费多少？

输入输出格式

输入格式：

 

数据的第一行有三个整数，n,m,kn,m,k，分别表示城市数，航线数和免费乘坐次数。
第二行有两个整数，s,ts,t，分别表示他们出行的起点城市编号和终点城市编号。
接下来有m行，每行三个整数，a,b,ca,b,c，表示存在一种航线，能从城市aa到达城市bb，或从城市bb到达城市aa，价格为cc。

 

输出格式：

 

只有一行，包含一个整数，为最少花费。

 

输入输出样例

输入样例#1： 复制

5 6 1
0 4
0 1 5
1 2 5
2 3 5
3 4 5
2 3 3
0 2 100

输出样例#1： 复制

8

说明

对于30%的数据,2 \le n \le 50,1 \le m \le 300,k=02≤n≤50,1≤m≤300,k=0;
对于50%的数据,2 \le n \le 600,1 \le m \le 6000,0 \le k \le 12≤n≤600,1≤m≤6000,0≤k≤1;
对于100%的数据,2 \le n \le 10000,1 \le m \le 50000,0 \le k \le 102≤n≤10000,1≤m≤50000,0≤k≤10,0 \le s,t<n,0 \le a,b<n,a\neq b,0 \le c \le 10000≤s,t<n,0≤a,b<n,a≠b,0≤c≤1000

2018.12.10 增加一组 hack 数据

 

解题思路：分层图最短路模板题

分层图就是把原来的图分增，上一层向下一层连边，每一次要么在当前层继续走，要么进入下一层，类似于bfs的思想。

第一种解法： 直接建立k层图，每层之间连边跑第一层到第k层的最短路。

第二种解法：利用堆优化的dij，将原来的点集分为两个，V1，V2，其中V1是已经确定的离远点最近的点，每次利用已经确定的点去更新未确定的点。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<bitset>
#include<map>
using namespace std;
#define LL long long
#define N 10005
#define maxn 1000005
#define inf 0x3f3f3f3f
#define sca(x) scanf("%d",&x)

struct node
{
    int to,w,nt;
}g[N*12];
int tot;
int head[N];

void addedg(int x,int y,int w)
{
    tot++;
    g[tot].to=y;
    g[tot].w=w;
    g[tot].nt=head[x];
    head[x]=tot;
}

struct Q
{
    int now,dist,cnt;
    friend bool operator <(Q x,Q y)
    {
        return x.dist>y.dist;//这里是大于号
    }
};

int dp[10005][11];
int vis[10005][11];
void dijstar(int s,int t,int k)
{
    memset(dp,inf,sizeof(dp));
    memset(vis,0,sizeof(vis));
    dp[s][0]=0;
    priority_queue<Q>q;
    Q t1;
    t1.now=s;t1.dist=0;t1.cnt=0;
    q.push(t1);
    while(!q.empty())
    {
        Q tmp=q.top();
        q.pop();
        if(vis[tmp.now][tmp.cnt])continue; //如果当前点已经确定就跳过。
        vis[tmp.now][tmp.cnt]=1;
        for(int i=head[tmp.now];i!=0;i=g[i].nt)
        {
            int v=g[i].to;
            if(tmp.cnt<k &&dp[v][tmp.cnt+1]>dp[tmp.now][tmp.cnt])//考虑使用优惠，则进入下一层
            {
                dp[v][tmp.cnt+1]=dp[tmp.now][tmp.cnt];
                q.push(Q{v,dp[v][tmp.cnt+1],tmp.cnt+1});
            }
            if(!vis[v][tmp.cnt] &&dp[v][tmp.cnt]>dp[tmp.now][tmp.cnt]+g[i].w)//不使用优惠则继续在当前层。
            {
                dp[v][tmp.cnt]=dp[tmp.now][tmp.cnt]+g[i].w;
                q.push(Q{v,dp[v][tmp.cnt],tmp.cnt});
            }
        }
    }
}

int main()
{
    int n,m,k;
    cin>>n>>m>>k;
    int s,t;
    cin>>s>>t;
    for(int i=1;i<=m;i++)
    {
        int x,y,w;
        sca(x),sca(y),sca(w);
        x++,y++;
        addedg(x,y,w);
        addedg(y,x,w);
    }
    dijstar(s+1,t+1,k);
    int ans=inf;
    for(int i=0;i<=k;i++)ans=min(ans,dp[t+1][i]);
    cout<<ans<<endl;
}