CodeForces - 1473E Minimum Path （分层图最短路）

题目链接：点击这里

题目大意：
给定一张 $n$ 个点， $m$ 条边的无向图，图中没有自环和重边。定义一条路径的权是路径上所有的边权之和减去所有边权最大值并加上所有边权的最小值。对每个点 $i\in [2,n]$ 求出从 $1$ 到i的路径权最小值

题目分析：
距离的定义为 $\sum w_i-max+min$ ，其中 $\sum w_i$ 就是最短路，减最大值和加最小值可以看成最短路的附加条件，即作为分层图的分层条件
设 $dis[0][],dis[1][],dis[2][],dis[3][]$ 分别表示原图最短路，去掉最大值的最短路，增加了最小值的最短路，去掉最大值且增加了最小值的最短路，初始时直接加入第 $0$ 层和第 $3$ 层的原点跑分层图即可
注意此题会爆 $int$

具体细节见代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<set>
#include<map>
#define ll long long
#define inf 0x3f3f3f3f
#define int ll 
using namespace std;
int read()
{
	int res = 0,flag = 1;
	char ch = getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch == '-') flag = -1;
		ch = getchar();
	}
	while(ch>='0' && ch<='9')
	{
		res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
		ch = getchar();
	}
	return res*flag;
}
const int maxn = 4e5+5;
const int mod = 1e9+7;
const double pi = acos(-1);
const double eps = 1e-8;
struct Edge{
	int nxt,to,val;
}edge[maxn];
int n,m,k,head[maxn],cnt,dis[5][maxn];
bool vis[5][maxn];
void addedge(int from,int to,int val)
{
	edge[++cnt].nxt = head[from];
	edge[cnt].to = to;
	edge[cnt].val = val;
	head[from] = cnt;
}
void dijkstra(int u)
{
	priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>>qu;
	memset(dis,127,sizeof(dis));
	dis[0][u] = dis[3][u] = 0;
	qu.push(make_pair(0,u));
	qu.push(make_pair(0,u+3*(n+1)));
	while(!qu.empty())
	{
		int h = qu.top().second;
		qu.pop();
		int level = h/(n+1); h %= (n+1);
		if(vis[level][h]) continue;
		vis[level][h] = true;
		for(int i = head[h];i;i = edge[i].nxt)
		{
			int to = edge[i].to,val = edge[i].val;
			if(dis[level][to] > dis[level][h]+val)
			{
				dis[level][to] = dis[level][h]+val;
				qu.push(make_pair(dis[level][to],to+level*(n+1)));
			}
			if(level == 3) continue;
			if(!level)
			{
				if(dis[level+1][to] > dis[level][h])
				{
					dis[level+1][to] = dis[level][h];
					qu.push(make_pair(dis[level+1][to],to+(level+1)*(n+1)));
				}
				if(dis[level+2][to] > dis[level][h]+val+val)
				{
					dis[level+2][to] = dis[level][h]+val+val;
					qu.push(make_pair(dis[level+2][to],to+(level+2)*(n+1)));
				} 
			}
			else if(level == 1){
				if(dis[level+2][to] > dis[level][h]+val+val)
				{
					dis[level+2][to] = dis[level][h]+val+val;
					qu.push(make_pair(dis[level+2][to],to+(level+2)*(n+1)));
				}
			}
			else {
				if(dis[level+1][to] > dis[level][h])
				{
					dis[level+1][to] = dis[level][h];
					qu.push(make_pair(dis[level+1][to],to+(level+1)*(n+1)));
				}
			}
		}
	}
}

signed main()
{
	n = read(),m = read();
	for(int i = 1;i <= m;i++)
	{
		int from = read(),to = read(),val = read();
		addedge(from,to,val);
		addedge(to,from,val);
	}
	dijkstra(1);
	for(int i = 2;i <= n;i++) printf("%lld%c",dis[3][i],i==n ? '\n' : ' ');
	return 0;
}