CF1473E. Minimum Path
题意:
给你一个无向图,一条长度为 k k k的路径 e 1 , e 2 , ⋯ , e k e_1,e_2,\cdots,e_k e1,e2,⋯,ek的权值为 ∑ i = 1 k w e i − max i = 1 k w e i + min i = 1 k w e i \sum\limits_{i=1}^k w_{e_i}-\max\limits_{i=1}^{k} w_{e_i}+\min\limits_{i=1}^{k} w_{e_i} i=1∑kwei−i=1maxkwei+i=1minkwei,问从 1 1 1到其他节点权值最小为多少。
思路:
题目可以等同于从
1
1
1到
i
i
i的路径中,一条路权值变成
0
0
0,一条路权值加倍,求最短路,可以想到分层图。
d
i
s
[
i
]
[
0
/
1
]
[
0
/
1
]
dis[i][0/1][0/1]
dis[i][0/1][0/1]表示到
i
i
i节点,一条边权值变不变成
0
0
0,一条边权值加不加倍的最短路。
因为求最短路,所以可以保证去掉的是最大边,加倍的是最小边。
最后输出
d
i
s
[
i
]
[
1
]
[
1
]
dis[i][1][1]
dis[i][1][1]即可。
代码:
#include<bits/stdc++.h>
#define pii pair<int,int>
#define int long long
#define cl(x,y) memset(x,y,sizeof(x))
#define loop(x,y,z) for(x=y;x<=z;x++)
#define reve(x,y,z) for(x=y;x>=z;x--)
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define all(x) x.begin(),x.end()
#define lson x<<1,l,mid
#define rson x<<1|1,mid+1,r
#define INF 1e18
const int N=1e6+10;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
const double eps=1e-8;
const double pi=acos(-1);
using namespace std;
struct edge
{
int u,v,w;
}e[N];
int head[N],len=0,dis[N][2][2],vis[N][2][2]={0};
struct node
{
int u,w,x,y;
friend bool operator < (node a,node b)
{
return a.w>b.w;
}
};
void add(int u,int v,int w)
{
e[++len]={head[u],v,w};
head[u]=len;
}
void dij()
{
int i;
priority_queue<node> q;
q.push({1,0,0,0});
while(!q.empty())
{
node pre=q.top();
q.pop();
int u=pre.u,x=pre.x,y=pre.y;
if(vis[u][x][y])
continue;
vis[u][x][y]=1;
for(i=head[u];i;i=e[i].u)
{
int v=e[i].v,w=e[i].w;
if(dis[v][x][y]>dis[u][x][y]+w)
{
dis[v][x][y]=dis[u][x][y]+w;
q.push({v,dis[v][x][y],x,y});
}
if(x==0)
{
if(dis[v][1][y]>dis[u][x][y])
{
dis[v][1][y]=dis[u][x][y];
q.push({v,dis[v][1][y],1,y});
}
}
if(y==0)
{
if(dis[v][x][1]>dis[u][x][y]+2*w)
{
dis[v][x][1]=dis[u][x][y]+2*w;
q.push({v,dis[v][x][1],x,1});
}
}
}
}
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int n,m,i;
cin>>n>>m;
cl(dis,inf);
dis[1][0][0]=0;
for(i=1;i<=m;i++)
{
int u,v,w;
cin>>u>>v>>w;
if(min(u,v)==1)
dis[max(u,v)][1][1]=w;
add(u,v,w);
add(v,u,w);
}
dij();
for(i=2;i<=n;i++)
cout<<dis[i][1][1]<<" ";
cout<<endl;
return 0;
}