P4568 飞行路线 (分层图最短路)

题目链接

题意：给你一张图，要你求出S到E的最短路，你可以使这张图中的k条边的值变成0.

思路：我们把这张图复制k+1份，让相邻的两层之间连单向边，并且让边权值为0。​​​​​​最后设置一个super点，把每一层的终点都汇入这个点，注意也是单向边哦。用dijkstra求出最短路，那么答案就是d【super】。

在写这个题的时候发现了骚操作，就是dijkstra的模板可以再改一下，就是没必要写结构体了，可以直接用pair，优先队列套pair，默认的优先级是，按第一个大排序，第一个相同，按第二个大排序。

#include <bits/stdc++.h>
#define ll long long
#define pi pair<int,int>
#define mk make_pair
#define pb push_back
using namespace std;
const int maxn = 2e5+10;
int n,m,K;
vector<pi>G[maxn];
int num(int k,int x)
{
	return k*n+x;
}
int d[maxn],vis[maxn];
int dij(int s,int t)
{
	memset(d,0x3f,sizeof(d));
	memset(vis,0,sizeof(vis));
	priority_queue<pi>q;
	q.push(mk(0,s));
	d[s] = 0;
	while(!q.empty())
	{
		int u = q.top().second;
		q.pop();
		if(vis[u])continue;
		vis[u] = 1;
		for(int i=0;i<G[u].size();i++)
		{
			int v = G[u][i].first;
			int w = G[u][i].second;
			if(d[v] > d[u] + w)
			{
				d[v] = d[u] + w;
				q.push(mk(-d[v],v));
			}
		}
	}
}
int main()
{
	scanf("%d%d%d",&n,&m,&K);
	int s,t;
	scanf("%d%d",&s,&t);
	for(int i=1;i<=m;i++)
	{
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		G[u].pb(mk(v,w));
		G[v].pb(mk(u,w));
		for(int k = 1;k<=K;k++)
		{
			G[num(k,u)].pb(mk(num(k,v),w));
			G[num(k,v)].pb(mk(num(k,u),w));
			G[num(k-1,u)].pb(mk(num(k,v),0));
			G[num(k-1,v)].pb(mk(num(k,u),0));
		}
	} 
	
	int sp = (K+1)*n;
	for(int i=0;i<=K;i++) {
		G[num(i,t)].pb(mk(sp,0));
	}
	dij(s,t);
	printf("%d\n",d[sp]);
	return 0;
}