2763: [JLOI2011]飞行路线
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 2636 Solved: 998
[ Submit][ Status][ Discuss]
Description
Alice和Bob现在要乘飞机旅行,他们选择了一家相对便宜的航空公司。该航空公司一共在n个城市设有业务,设这些城市分别标记为0到n-1,一共有m种航线,每种航线连接两个城市,并且航线有一定的价格。Alice和Bob现在要从一个城市沿着航线到达另一个城市,途中可以进行转机。航空公司对他们这次旅行也推出优惠,他们可以免费在最多k种航线上搭乘飞机。那么Alice和Bob这次出行最少花费多少?
Input
数据的第一行有三个整数,n,m,k,分别表示城市数,航线数和免费乘坐次数。
第二行有两个整数,s,t,分别表示他们出行的起点城市编号和终点城市编号。(0<=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
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<=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.
Source
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<stack>
#include<bitset>
#include<ext/pb_ds/priority_queue.hpp>
using namespace std;
const int maxn = 2E4 + 10;
const int T = 10;
struct E{
int to,w; E(){}
E(int to,int w): to(to),w(w){}
};
int n,m,k,tot,s,t,dis[maxn*T],num[maxn][11];
bool vis[maxn*T];
vector <E> v[maxn*T];
queue <int> Q;
int main()
{
#ifdef DMC
freopen("DMC.txt","r",stdin);
#endif
cin >> n >> m >> k >> s >> t;
memset(dis,127,sizeof(dis));
for (int i = 0; i < n; i++)
for (int j = 0; j <= k; j++) num[i][j] = tot++;
while (m--)
{
int x,y,w; scanf("%d%d%d",&x,&y,&w);
for (int i = 0; i <= k; i++)
{
int A = num[x][i],B = num[y][i];
v[A].push_back(E(B,w));
v[B].push_back(E(A,w));
if (i < k)
{
v[A].push_back(E(num[y][i+1],0));
v[B].push_back(E(num[x][i+1],0));
}
}
}
Q.push(num[s][0]); vis[num[s][0]] = 1; dis[num[s][0]] = 0;
while (!Q.empty())
{
int g = Q.front(); Q.pop(); vis[g] = 0;
for (int i = 0; i < v[g].size(); i++)
{
E e = v[g][i];
if (dis[e.to] > dis[g] + e.w)
{
dis[e.to] = dis[g] + e.w;
if (!vis[e.to]) vis[e.to] = 1,Q.push(e.to);
}
}
}
int Ans = ~0U>>1;
for (int i = 0; i <= k; i++) Ans = min(Ans,dis[num[t][i]]);
cout << Ans;
return 0;
}