Description
给定一个 n n n 个点, m m m 条边的无向图,每条边都有一个长度。你可以至多做 k k k 次操作,每次操作中选定两条不同的边并将它们的长度交换。
你需要求出,从 1 1 1 到 n n n 的最短路的最小值。
Solution
首先,假设我们已经找到了一条路径,我们必然会将路径上后 k k k 大的边替换为未选择的边中前 k k k 小的边。也就是说,存在某个阈值 L L L,若边权 ≤ L \le L ≤L 那么它一定被最终选择,若 > L >L >L 那么它可能会被中途替换掉。
枚举 L L L 并考虑 dp \text{dp} dp。
令 f u , i , j f_{u,i,j} fu,i,j 表示,目前走到了点 u u u,边权不超过 L L L 的边选了 i i i 条(不包含被换入的边),总共使用了 j j j 次魔法;在此前提下,所有边权超过 L L L 且被选定的边(不包含被换出的边)的长度之和的最小值。
对于某一条边 ( u , v , w ) (u,v,w) (u,v,w),它能产生下面的转移:
-
若 w ≤ L w \le L w≤L,那么 f u , i , j → f v , i + 1 , j f_{u,i,j} \to f_{v,i+1,j} fu,i,j→fv,i+1,j
-
若 w > L w>L w>L,
-
若不将此边替换,那么 f u , i , j + w → f v , i , j f_{u,i,j}+w \to f_{v,i,j} fu,i,j+w→fv,i,j
-
若替换此边,那么 f u , i , j → f v , i , j + 1 f_{u,i,j} \to f_{v,i,j+1} fu,i,j→fv,i,j+1
-
边界:
f
u
,
0
,
0
=
1
f_{u,0,0}=1
fu,0,0=1
答案:
f
n
,
i
,
j
+
∑
i
=
1
m
w
i
×
[
w
i
≤
L
]
f_{n,i,j}+\sum_{i=1}^m w_i \times [w_i \le L]
fn,i,j+∑i=1mwi×[wi≤L]
注意到这个 dp \text{dp} dp 是有后效性的,所以我们建立出一个包含 n m k nmk nmk 个点的图,在此图上用 Dijkstra 或 SPFA 跑最短路即可。
同时,我们并不需要在 [ 0 , 1 0 9 ] [0,10^9] [0,109] 中随意枚举 L L L,我们只需令 L L L 为各条边的边权即可。
总复杂度 O ( n m 2 k ) O(nm^2k) O(nm2k),注意剪枝。