题目
题目链接:
一个无向图,有
n
n
n个点,
m
m
m条边。为1到
x
x
x的边权。即减去最大的
w
w
w,加上两倍最小值。
分层最短路。
d
i
s
[
v
]
[
0
]
dis[v][0]
dis[v][0]表示初始状态,什么都没做
d
i
s
[
v
]
[
1
]
dis[v][1]
dis[v][1]表示已经做了减的操作
d
i
s
[
v
]
[
2
]
dis[v][2]
dis[v][2]表示做了加的操作
d
i
s
[
v
]
[
3
]
dis[v][3]
dis[v][3]表示减加都做了
答案就是
m
i
n
(
d
i
s
[
v
]
[
0
]
,
d
i
s
[
v
]
[
3
]
)
min(dis[v][0],dis[v][3])
min(dis[v][0],dis[v][3])
代码
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define int long long
using namespace std;
const int INF = 0x3f3f3f3f;
const int N = 2e5+5;
struct sut {
int to,next,val;
} edge[N<<1];
int dis[N][5],head[N];
bool vis[N][5];
int tot=0;
void addedge(int u,int v,int w)
{
edge[++tot].to=v;
edge[tot].val=w;
edge[tot].next=head[u];
head[u]=tot;
}
int n,m;
void dij(int s)
{
memset(dis,INF,sizeof dis);
dis[s][0]=0;
priority_queue<pair<int,int>> q;
q.push({0,s});
while(!q.empty()) {
int now=q.top().second;
q.pop();
int c=now/n;
now%=n;
if(vis[now][c]) continue;
vis[now][c]=1;
// cout<<now<<" "<<c<<endl;
int u=now;
for(int i=head[now]; i; i=edge[i].next) {
int v=edge[i].to;
if(dis[v][c]>dis[u][c]+edge[i].val) {
dis[v][c]=dis[u][c]+edge[i].val;
if(!vis[v][c]) {
q.push({-dis[v][c],v+c*n});
}
}
//c=1 - c=2 + c=3 ans
if(c!=1&&c!=3&&dis[v][c+1]>dis[u][c]) {
dis[v][c+1]=dis[u][c];
if(!vis[v][c+1]) {
q.push({-dis[v][c+1],v+(c+1)*n});
}
}
if(c!=2&&c!=3&&dis[v][c+2]>dis[u][c]+2*edge[i].val) {
dis[v][c+2]=dis[u][c]+2*edge[i].val;
if(!vis[v][c+2]) {
q.push({-dis[v][c+2],v+(c+2)*n});
}
}
}
}
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n>>m;
for(int i=1; i<=m; i++) {
int u,v,w;
cin>>u>>v>>w;
--u;//
--v;//注意点要从0开始算!!!!!
addedge(u,v,w);
addedge(v,u,w);
}
dij(0);
for(int i=1; i<n; i++) {
cout<<min(dis[i][0],dis[i][3])<<" ";
}
return 0;
}