飞行路线——分层图

 

飞行路线

题目描述

Alice 和 Bob 现在要乘飞机旅行，他们选择了一家相对便宜的航空公司。

该航空公司一共在 n 个城市设有业务，设这些城市分别标记为 0 到 n-1，一共有 m 种航线，每种航线连接两个城市，并且航线有一定的价格。

Alice 和 Bob 现在要从一个城市沿着航线到达另一个城市，途中可以进行转机。

航空公司对他们这次旅行也推出优惠，他们可以免费在最多 k 种航线上搭乘飞机。

那么 Alice 和 Bob 这次出行最少花费多少？

输入格式

第一行有三个整数，n,m,k，分别表示城市数，航线数和免费乘坐次数。

第二行有两个整数，s,t，分别表示他们出行的起点城市编号和终点城市编号。

接下来有 m 行，每行三个整数，a,b,c，表示存在一种航线，能从城市 a 到达城市 b，或从城市 b 到达城市 a，价格为 c。

输出格式

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

样例输入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≤n≤50, 1≤m≤300, k=0;
对于50%的数据, 2≤n≤600, 1≤m≤6000, 0≤k≤1;
对于100%的数据, 2≤n≤10000, 1≤m≤50000, 0≤k≤10.

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> PII;
const int N=1000003;//开大
int n,m,dis[N],state[N],k,s,t;
bool bw[N];
struct ed{
	int to,k; 
};
vector<ed>e[N];
void dj(){
	memset(dis,0x3f,sizeof(dis));
	priority_queue<PII,vector<PII>,greater<PII> > q;
	q.push((PII){0,s});
	dis[s]=0;
	while(!q.empty()){
		int x=q.top().second;
		q.pop();
		if(state[x])continue;
		state[x]=1;
		for(int i=0;i<e[x].size();i++){
			int to=e[x][i].to,w=e[x][i].k;
			if(dis[to]>dis[x]+w){
				dis[to]=dis[x]+w;
				q.push((PII){dis[to],to});
			}
		}
	}
}
int main(){
	scanf("%d%d%d",&n,&m,&k);
	scanf("%d%d",&s,&t);
	s++;t++;
	for(int i=1;i<=m;i++){
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		b++;a++;
		for(int j=0;j<=k;j++){
			e[a].push_back((ed){b,c});
			e[b].push_back((ed){a,c});
			if(j!=k){
				e[a].push_back((ed){b+n,0});
				e[b].push_back((ed){a+n,0});
			}		
			a+=n;
			b+=n;
		}
	}
	dj();
	int ans=INT_MAX;
	for(int i=0;i<=k;i++){
		ans=min(ans,dis[n*i+t]);
	}
	cout<<ans;
}

洛谷测得

N 开 1e4 91分

开 1e6 100分