题意
有一个
n
n
n个顶点
m
m
m条边的图中,每条边有边权,你有
k
k
k个任务,第
i
i
i个任务为从编号为
a
i
a_i
ai的点走到编号为
b
i
b_i
bi的点。
同时你又可以随时在当前节点打开一个传送门,在两个传送门之间可以瞬间传送。但是最多同时存在两个传送门,当你想要在已经有两个传送门的情况下打开传送门,就需要远程关闭一个传送门。
求从顶点
1
1
1开始,完成这
k
k
k个任务所经过的最小边权。
题解
这道题使用
d
p
dp
dp。
先使用
F
l
o
y
d
Floyd
Floyd算法直接求出多源最短路,用
d
i
s
[
i
]
[
j
]
dis[i][j]
dis[i][j]记录。
可以很容易想到设
d
p
[
i
]
[
v
]
[
a
]
[
b
]
dp[i][v][a][b]
dp[i][v][a][b],表示做任务
i
i
i时当前在顶点
v
v
v,存在两个位于顶点
a
a
a和
b
b
b的传送门时的最短距离。
由于每次选择创建一个传送门总比到原有传送门传送更优,所以每次只需要关心一个传送门,也就是可以删去一维为
d
p
[
i
]
[
v
]
[
p
]
dp[i][v][p]
dp[i][v][p]。
由于
n
,
k
≤
300
n,k\leq300
n,k≤300,此时复杂度为
O
(
k
n
3
)
O(kn^3)
O(kn3)仍会超时,所以还需要优化。
考虑将
[
v
]
[v]
[v]这一维删去,将任务拆成到达两个节点,用
c
[
i
]
c[i]
c[i]记录。
这样就可以设成
d
p
[
i
]
[
p
]
dp[i][p]
dp[i][p],
c
[
i
]
c[i]
c[i]为当前顶点。
转移有两种情况:
- 直接从 c [ i − 1 ] c[i-1] c[i−1]走到 c [ i ] c[i] c[i]: d p [ i ] [ p ] = m i n ( d p [ i ] [ p ] , d p [ i − 1 ] [ p ] + d i s [ c [ i − 1 ] ] [ c [ i ] ] ) ; dp[i][p]=min(dp[i][p],dp[i-1][p]+dis[c[i-1]][c[i]]); dp[i][p]=min(dp[i][p],dp[i−1][p]+dis[c[i−1]][c[i]]);
- 先走到顶点 q q q,设置传送门到 p p p,在从 p p p走到 c [ i ] c[i] c[i]或先传送到 p p p,从 p p p走到 q q q设置传送门,再从 q q q走到 c [ i ] c[i] c[i](最后传送门在 p p p还是在 q q q不确定,都要转移): d p [ i ] [ p ] = m i n ( d p [ i ] [ p ] , d p [ i − 1 ] [ p ] + m i n ( d i s [ c [ i − 1 ] ] [ q ] + d i s [ p ] [ c [ i ] ] , d i s [ p ] [ q ] + d i s [ q ] [ c [ i ] ] ) ) ; dp[i][p]=min(dp[i][p],dp[i-1][p]+min(dis[c[i-1]][q]+dis[p][c[i]],dis[p][q]+dis[q][c[i]])); dp[i][p]=min(dp[i][p],dp[i−1][p]+min(dis[c[i−1]][q]+dis[p][c[i]],dis[p][q]+dis[q][c[i]])); d p [ i ] [ q ] = m i n ( d p [ i ] [ q ] , d p [ i − 1 ] [ p ] + m i n ( d i s [ c [ i − 1 ] ] [ q ] + d i s [ p ] [ c [ i ] ] , d i s [ p ] [ q ] + d i s [ q ] [ c [ i ] ] ) ) ; dp[i][q]=min(dp[i][q],dp[i-1][p]+min(dis[c[i-1]][q]+dis[p][c[i]],dis[p][q]+dis[q][c[i]])); dp[i][q]=min(dp[i][q],dp[i−1][p]+min(dis[c[i−1]][q]+dis[p][c[i]],dis[p][q]+dis[q][c[i]]));
初始化:
d
p
[
0
]
[
1
]
=
0
dp[0][1]=0
dp[0][1]=0,其他全部赋值为
i
n
f
inf
inf;
c
[
0
]
=
1
c[0]=1
c[0]=1。
(其实是假设第
0
0
0个任务为到达顶点
1
1
1,并在原地设置传送门)
注意,可能需要使用 l o n g l o n g long\ long long long。
时间复杂度 O ( k × n 2 ) O(k\times n^2) O(k×n2)。
#include<bits/stdc++.h>
using namespace std;
long long dis[400][400];
long long dp[800][400];
int c[800];
int main(){
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) dis[i][j]=1ll<<40;
for(int i=0;i<=k*2;i++) for(int j=1;j<=n;j++) dp[i][j]=1ll<<40;
for(int i=1;i<=n;i++) dis[i][i]=0;
for(int i=1;i<=m;i++){
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
dis[x][y]=dis[y][x]=min(dis[x][y],1ll*z);
}
for(int l=1;l<=n;l++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
dis[i][j]=min(dis[i][j],dis[i][l]+dis[l][j]);
for(int i=1;i<=k;i++)
scanf("%d%d",&c[i*2-1],&c[i*2]);
k*=2;
dp[0][1]=0;
c[0]=1;
for(int i=1;i<=k;i++){//dp,j层枚举p,l层枚举q
for(int j=1;j<=n;j++){
dp[i][j]=min(dp[i][j],dp[i-1][j]+dis[c[i-1]][c[i]]);
for(int l=1;l<=n;l++){
dp[i][j]=min(dp[i][j],dp[i-1][j]+min(dis[c[i-1]][l]+dis[j][c[i]],dis[j][l]+dis[l][c[i]]));
dp[i][l]=min(dp[i][l],dp[i-1][j]+min(dis[c[i-1]][l]+dis[j][c[i]],dis[j][l]+dis[l][c[i]]));
}
}
}
long long minn=1ll<<50;
for(int i=1;i<=n;i++) minn=min(minn,dp[k][i]);
printf("%lld\n",minn);
return 0;
}