题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=6166
题目大意: 给出一个n个点m条边的有向图, 询问k个点中两两最短路的最小值。 ( n,m,k≤105 )
思路1: 先建k个新点,依次复制k个点的出边, 用这k个新点做多源最短路, 由于这样做可能得到自己到自己的非法答案, 所以最短路时需要维护两种起点不同的最短路和次短路。
思路2:考虑最终的答案是u->v的最短路, 对于两个不同的编号, 它们的二进制至少又一个位上不同的, 按每个二进制位分组跑log次集合到集合的最短路即可。 所谓集合到集合的最短路, 新建一个超级起点连一条指向所有起点费用为0的边, 让所有终点连一条向超级终点的费用为0的边跑普通的最短路即可。
PS: 发现自己最短路根本不过关, 竟然连spfa, dijkstra这种其实可以求多源最短路都不知道TAT。
#include <queue>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#define ll long long
const int N = (int)2e5 + 10;
const ll inf = 1LL << 60;
using namespace std;
int n, m, k;
int cnt, lst[N], nxt[N], to[N], w[N]; bool key[N]; ll ans;
void add(int u, int v, int c){
nxt[++ cnt] = lst[u]; lst[u] = cnt; to[cnt] = v; w[cnt] = c;
}
ll dis[N][2]; int pre[N][2];
int head, tail, que[N * 100]; bool in[N];
void spfa(){
for (int i = 1; i <= 2 * n; i ++) dis[i][0] = dis[i][1] = inf;
head = tail = 0;
for (int u = 1; u <= n; u ++)
if (key[u]){
for (int j = lst[u]; j; j = nxt[j]){
int v = to[j];
add(u + n, v, w[j]);
}
dis[u + n][0] = 0; pre[u + n][0] = u;
que[++ tail] = u + n; in[u + n] = 1;
}
while (head < tail){
int u = que[++ head]; in[u] = 0;
for (int j = lst[u]; j; j = nxt[j]){
int v = to[j];
if (dis[v][0] > dis[u][0] + w[j]){
if (pre[v][0] != pre[u][0]){
dis[v][1] = dis[v][0];
pre[v][1] = pre[v][0];
}
dis[v][0] = dis[u][0] + w[j];
pre[v][0] = pre[u][0];
if (!in[v]) in[que[++ tail] = v] = 1;
}
else if (dis[v][1] > dis[u][0] + w[j] && pre[v][0] != pre[u][0]){
dis[v][1] = dis[u][0] + w[j];
pre[v][1] = pre[u][0];
if (!in[v]) in[que[++ tail] = v] = 1;
}
if (dis[v][1] > dis[u][1] + w[j] && pre[v][0] != pre[u][1]){
dis[v][1] = dis[u][1] + w[j];
pre[v][1] = pre[u][1];
if (!in[v]) in[que[++ tail] = v] = 1;
}
}
}
for (int i = 1; i <= n; i ++)
if (key[i]){
if (pre[i][0] != i) ans = min(ans, dis[i][0]);
else ans = min(ans, dis[i][1]);
}
}
int main(){
int cas = 0; int T; scanf("%d", &T);
while (T --){
scanf("%d %d", &n, &m);
printf("Case #%d: ", ++ cas);
cnt = 0;
for (int i = 1; i <= 2 * n; i ++) lst[i] = 0, key[i] = 0;
for (int i = 1, u, v, c; i <= m; i ++){
scanf("%d %d %d", &u, &v, &c);
add(u, v, c);
}
scanf("%d", &k);
for (int i = 1, x; i <= k; i ++){
scanf("%d", &x); key[x] = 1;
}
ans = inf;
spfa();
printf("%lld\n", ans);
}
return 0;
}