【JLOI2015 DAY2】通道连接

问题描述

小铭铭最近进入了某情报部门，该部门正在被如何建立安全的通道连接困扰。
该部门有 n 个情报站，用 1 到 n 的整数编号。给出 m 对情报站 ui;vi 和费用 wi，表示情报站 ui 和 vi 之间可以花费 wi 单位资源建立通道。
如果一个情报站经过若干个建立好的通道可以到达另外一个情报站，那么这两个情报站就建立了通道连接。形式化地，若 ui 和 vi 建立了通道，那么它们建立了通道连接；若 ui 和 vi 均与 ti 建立了通道连接，那么 ui 和 vi 也建立了通道连接。
现在在所有的情报站中，有 p 个重要情报站，其中每个情报站有一个特定的频道。小铭铭面临的问题是，需要花费最少的资源，使得任意相同频道的情报站之间都建立通道连接。

输入格式

第一行包含三个整数 n;m;p，表示情报站的数量，可以建立的通道数量和重要情报站的数量。接下来 m 行，每行包含三个整数
ui;vi;wi，表示可以建立的通道。最后有 p 行，每行包含两个整数 ci;di，表示重要情报站的频道和情报站的编号。

输出格式

输出一行一个整数，表示任意相同频道的情报站之间都建立通道连接所花费的最少资源总量。

样例输入

5 8 4
1 2 3
1 3 2
1 5 1
2 4 2
2 5 1
3 4 3
3 5 1
4 5 1
1 1
1 2
2 3
2 4

样例输出

4

提示

选择 (1; 5); (3; 5); (2; 5); (4; 5) 这 4 对情报站连接。
对于 100% 的数据，0 < ci <= p <= 10; 0 < ui;vi;di <= n <= 1000; 0 <= m <= 3000; 0 <= wi <=20000。

题解

题目大意为给出几个特殊点，求把它们连接起来的最小值。特殊点数目很少，显然会用到斯坦纳树。
本题特殊的地方在于，只要类型相同的特殊点在一个联通块里即可，并不需要所有的点都被联通。那我们对每一种类型都建一棵斯坦纳树？当然不行，有的边可能会被重复计算。
我们直接跑一遍斯坦纳树的版子，这样就求出了在每个点构建斯坦纳树的最小值，然后再把特殊点的类型状态压缩，讨论点集之间是否联通的情况，同时也避免了计算重复的情况。

代码

#include<stdio.h>
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
const int maxn=1e3+5;
const int inf=1e9;
int n,m,p,u,v,w,cnt,tot,c[15],d[15],cs[15],Last[maxn];
int g[maxn+20],f[maxn][maxn+20];
bool flag[maxn];
queue<int> q;
struct node
{
    int End,Next,Len;
}edge[6*maxn];
void Addedge(int x,int y,int z)
{
    edge[++cnt].End=y,edge[cnt].Len=z;
    edge[cnt].Next=Last[x],Last[x]=cnt;
}
void spfa(int S)
{
    while(!q.empty())
    {
        int x=q.front();
        flag[x]=false,q.pop();
        for(int i=Last[x];i;i=edge[i].Next)
        {
            int y=edge[i].End;
            if(f[y][S]>f[x][S]+edge[i].Len)
            {
                f[y][S]=f[x][S]+edge[i].Len;
                if(!flag[y]) q.push(y),flag[y]=true;
            }
        }
    }
}
int main()
{
    scanf("%d%d%d",&n,&m,&p);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&u,&v,&w);
        Addedge(u,v,w),Addedge(v,u,w);
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<(1<<p);j++) f[i][j]=inf;
    for(int i=1;i<=p;i++)
    {
        scanf("%d%d",&c[i],&d[i]),tot=max(tot,c[i]);
        f[d[i]][1<<(i-1)]=0,cs[c[i]]|=1<<(i-1);
    }
    for(int S=1;S<(1<<p);S++)
    {
        for(int i=1;i<=n;i++)
        {
            for(int T=S;T;T=S&(T-1)) f[i][S]=min(f[i][S],f[i][T]+f[i][S^T]);
            if(f[i][S]<inf) flag[i]=true,q.push(i);
        }
        spfa(S);
    }
    for(int i=1;i<(1<<p);i++) g[i]=inf;
    for(int S=1;S<(1<<tot);S++)//对频道类型的状压讨论
    {
        int T=0;
        for(int i=0;i<tot;i++)
            if(S&(1<<i)) T|=cs[i+1];
        for(int i=1;i<=n;i++) g[T]=min(g[T],f[i][T]);
    }
    for(int S=1;S<(1<<p);S++)
        for(int T=S;T;T=S&(T-1)) g[S]=min(g[S],g[T]+g[S^T]);
    printf("%d\n",g[(1<<p)-1]);
    return 0;
}