「BZOJ2763」[JLOI2011]飞行路线【分层图最短路】

2763: [JLOI2011]飞行路线

Time Limit: 10 Sec Memory Limit: 128 MB Submit: 6174 Solved: 2498 [Submit][Status][Discuss]

Description

$Alice$和$Bob$现在要乘飞机旅行，他们选择了一家相对便宜的航空公司。该航空公司一共在$n$个城市设有业务，设这些城市分别标记为$0$到$n-1$，一共有$m$种航线，每种航线连接两个城市，并且航线有一定的价格。$Alice$和$Bob$现在要从一个城市沿着航线到达另一个城市，途中可以进行转机。航空公司对他们这次旅行也推出优惠，他们可以免费在最多$k$种航线上搭乘飞机。那么$Alice$和$Bob$这次出行最少花费多少？

Input

数据的第一行有三个整数，$n,m,k$，分别表示城市数，航线数和免费乘坐次数。
第二行有两个整数，$s,t$，分别表示他们出行的起点城市编号和终点城市编号。$(0\le s,t<n)$
接下来有m行，每行三个整数，$a,b,c$，表示存在一种航线，能从城市$a$到达城市$b$，或从城市$b$到达城市$a$，价格为$c$。$(0<=a,b<n,a$与$b$不相等，$0<=c<=1000)$

Output

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

Sample Input

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

Sample Output

8

HINT

对于$30\%$的数据,$2\le n\le 50,1\le m\le 300,k=0;$

对于$50\%$的数据,$2\le n\le 600,1\le m\le 6000,0\le k\le 1;$

对于$100\%$的数据,$2\le n\le 10000,1\le m\le 50000,0\le k\le 10.$

题解

• 多校过来的渣渣还债，分层图最短路模版题
• 分层图就是先把一张图$copy$ $k$张下来，图与图之间的连边规则是：对于第$i$张【每张图从上至下依次编号1～$k$】的图上的边$(u_1,v_1,w_1)$，与其对应的第二张图上的边$(u_2,v_2,w_2)$，连边$(u_1,v_2,w_3)$及$(v_1,u_2,w_4)$，其中$w_3$与$w_4$根据题意赋值，此题设为$0$，然后跑$dijkstra$就行了
• 下图为无向图建图方式
  

代码

#include<bits/stdc++.h>
 
using namespace std;
const int maxn=10005;
 
int u,v,w,n,m,s,t,k,dis[maxn*20];
 
struct node{
    int to,w;
    node(int a=0,int b=0) {
        to=a;w=b;
    }
};
vector<node> vec[maxn*20];
 
 
int dijkstra(int s,int t)
{
    priority_queue<pair<int,int> ,vector<pair<int,int> >,greater<pair<int,int> > > que;
    memset(dis,0x3f,sizeof(dis));
    dis[s]=0;
    que.push(pair<int,int>(0,s));
    while(!que.empty()){
        pair<int,int> cur=que.top();que.pop();
        if(dis[cur.second]<cur.first) continue;
        for(int i=0;i<vec[cur.second].size();i++){
            node nxt=vec[cur.second][i];
            if(dis[nxt.to]>dis[cur.second]+nxt.w){
                dis[nxt.to]=dis[cur.second]+nxt.w;
                que.push(pair<int,int>(dis[nxt.to],nxt.to));
            }
        }
    }
    return dis[t];
}
 
 
void add_edge(int u,int v,int w)
{
    vec[u].push_back(node(v,w));
}
 
int main()
{
    scanf("%d %d %d %d %d",&n,&m,&k,&s,&t);s++;t++;
    for(int i=1;i<=m;i++) {
        scanf("%d %d %d",&u,&v,&w);u++;v++;
        add_edge(u,v,w);add_edge(v,u,w);
        for(int j=1;j<=k;j++) {
            add_edge(j*n+u,j*n+v,w);add_edge(j*n+v,j*n+u,w);
            add_edge((j-1)*n+u,j*n+v,0);add_edge((j-1)*n+v,j*n+u,0);
        }
    }
    printf("%d\n",dijkstra(s,k*n+t));
 
}