GalaxyOJ-768 (重建图+SPFA)

题目

也可以在这里看到这题 https://nanti.jisuanke.com/t/15969

Problem Description

百度地图上有 n 个城市，城市编号依次为 1 到 n。地图中有若干个城市群，编号依次为 1 到 m。每个城市群包含一个或多个城市；每个城市可能属于多个城市群，也可能不属于任何城市群。

地图中有两类道路。第一类道路是 城市之间的快速路，两个城市 u,v 之间增加一条距离为 c 的边；第二类道路是 城市群之间的高速路，连接两个城市群 a,b，通过这条高速路，城市群 a 里的每个城市与城市群b 里的每个城市之间两两增加一条距离为 c 的边。图中所有边均为无向边。

你需要计算从城市 s 到城市 t 的最短路。

Input

第一行输入n, m，分别表示城市总数和城市群总数。

接下来一共输入 m 行。

第 i 行首先输入一个k[i] (1≤k[i]≤n)，表示第 i个城市群中的城市数为k[i]。接下来输入 k[i]个数，表示第 i 个城市群中每个城市的编号（保证一个城市群内的城市编号不重复且合法, ∑k[i]≤20000）

下一行输入一个整数m1 (0≤m[1]≤n)，表示有 m1条第一类道路，即城市之间的快速路。

接下来 m1行，每行输入三个整数u[i] ,v[i](1≤u[i] ,v[i]≤n),c[i] (1≤c[i]≤10^6 )，分别表示快速路连接的两个城市编号和边的距离。

下一行输入一个整数 m2(0≤m2 ≤m)，表示有 m2条第二类道路，即城市群之间的高速路。

接下来 m​2行，每行输入三个整数 a[i] ,b[i] (1≤a[i] ,b[i]≤m),l[i] (1≤l[i]≤10​^6​​ )，分别表示快速路连接的两个城市群编号和边的距离。

最后一行输入 s,t(1≤s,t≤n)，表示起点和终点城市编号。

2<=n<=20000,0<=m<=20000

Output

输出一个整数，表示城市 s 到城市 t 到最短路。如果不存在路径，则输出-1。

Sample Input

5 4
2 5 1
2 2 4
1 3
2 3 4
2
1 2 9
1 5 18
2
1 2 6
1 3 10
1 5

Sample Output

12

Hint

对于样例，1 -> 2 - > 5或者1 -> 4 -> 5是最短的路径，总长度为 12。

分析

• 这道题大意就是说一个图，点和点之间可以有边，然后有一些点群（群间的点可以重复），两个群之间若有条“特殊边”，那么这个群中的点都可以走这条“特殊边”到它所连着的群中的点，求起点到终点的最短路程。
• 咋一看，有种暴力方法：每次读入一个连接两个群的“特殊边”时，把这两个群中的点一一对应都练条边，然后来遍普通的最短路（SPFA即可）求出最短路程即可。
• 可是这样边数巨多，数据范围让我们面临着爆空间的风险，于是我们就想到了修改一下这个图。
• 为每一个群新建两个点来代表群中的点，然后为这两个代表点到群中每个点添加一条长度为0的边，然后群间相连的边只需要连在这几个代表点就行了。
• 至于为什么每个群要新建两个点，是为了让群中的点间的距离不变，一个点作为“群入点”，与群中每个点连一条起点为这个代表点的边，另一个点作为“群出点”，为群中每个点连个终点为代表点的边。若只用一个点作代表点，向群中的点连距离为0的无向边，那么群中的点两两间距离都变成0了，改变了原来的信息。
• 我的程序中，编号为x的群的群入点为n+x*2，出点为n+x*2+1。
• 附上张样例的图修改后的图（[]中的是起点，括号中是每个它出发的边的终点和其权值）
  

程序

#include <cstdio>
#include <cstring>
using namespace std;
#define For(x) for(int h=head[x],o=V[h],w=W[h]; h; h=to[h],o=V[h],w=W[h])
#define add(x,y,z) (to[++num]=head[x],head[x]=num,V[num]=y,W[num]=z)
int head[100000],to[100000],V[100000],W[100000],num;
int q[100000],d[100000],f[100000],st,ed;
int S,T,n,m;

void SPFA(){
    q[st=ed=0]=S;
    memset(d,0x7f,sizeof(d));
    d[S]=0;
    f[S]=1;
    while (st<=ed){
        int x=q[st];
        For(x) {
            if (w+d[x]<d[o]){   //用 x 松弛o 
                d[o]=w+d[x];
                if (!f[o]){
                    q[++ed]=o;
                    f[o]=1;
                } 
            }
        }
        f[x]=0;
        st++;
    }
}

int main(){
    scanf("%d%d",&n,&m);
    for (int i=1,o; i<=m; i++){
        scanf("%d",&o);
        for (int j=1,jj; j<=o; j++){
            scanf("%d",&jj);
            add(n+i*2,jj,0);        //入 
            add(jj,n+i*2+1,0);      //出 
        }
    }
    scanf("%d",&m);
    for (int u,v,w; m; m--){
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
        add(v,u,w);
    }
    scanf("%d",&m);
    for (int u,v,w; m; m--){
        scanf("%d%d%d",&u,&v,&w);
        add(n+u*2+1,n+v*2,w);
        add(n+v*2+1,n+u*2,w);
    }
    scanf("%d%d",&S,&T);

    /*输出图的信息
    for (int i=1; i<=n+5*2; i++,puts("")){
        printf("[%d]\t",i);
        For(i) printf("(%d %d) ",o,w);
    }
    */

    SPFA();
    if (d[T]!=d[0]) printf("%d",d[T]); else puts("-1");
}

提示

• 也是第一次见到这种题，长见识了。