bzoj 4006: [JLOI2015]管道连接

Description

小铭铭最近进入了某情报部门，该部门正在被如何建立安全的通道连接困扰。

该部门有 n 个情报站，用 1 到 n 的整数编号。给出 m 对情报站 ui;vi 和费用 wi，表示情

报站 ui 和 vi 之间可以花费 wi 单位资源建立通道。

如果一个情报站经过若干个建立好的通道可以到达另外一个情报站，那么这两个情报站就

建立了通道连接。形式化地，若 ui 和 vi 建立了通道，那么它们建立了通道连接；若 ui 和 vi 均

与 ti 建立了通道连接，那么 ui 和 vi 也建立了通道连接。

现在在所有的情报站中，有 p 个重要情报站，其中每个情报站有一个特定的频道。小铭铭

面临的问题是，需要花费最少的资源，使得任意相同频道的情报站之间都建立通道连接。

Input

第一行包含三个整数 n;m;p，表示情报站的数量，可以建立的通道数量和重要情报站的数

量。接下来 m 行，每行包含三个整数 ui;vi;wi，表示可以建立的通道。最后有 p 行，每行包含

两个整数 ci;di，表示重要情报站的频道和情报站的编号。

Output

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

Sample Input

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

Sample Output

4

HINT

选择 (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。

 

 

题解:

斯坦纳树变式,只需正常跑一边斯坦纳然后再加特判,这题中F[i][k]中第二维k不再表示k这个状态联通

而表示k状态中,相同频道的站都联通,所以我们最后再合并,避免漏解情况

可以理解为:斯坦纳跑出来的k是两个块用一些边连起来以后的联通块,而此题中一些边可以去掉,只需满足相同频道联通即可

 #include <algorithm>
 #include <iostream>
 #include <cstdlib>
 #include <cstring>
 #include <cstdio>
 #include <cmath>
 using namespace std;
 const int N=,M=;
 int gi(){
     int str=;char ch=getchar();
     while(ch>'' || ch<'')ch=getchar();
     while(ch>='' && ch<='')str=(str<<)+(str<<)+ch-,ch=getchar();
     return str;
 }
 int head[N],num=;
 struct Lin{
     int next,to,dis;
 }a[M<<];
 void init(int x,int y,int z){
     a[++num].next=head[x];a[num].to=y;a[num].dis=z;head[x]=num;
 }
 int n,m,col,f[N][<<],P=,tot,INF;
 bool vis[N];int q[N*],mod=N*;
 void spfa(int k){
     int t=,sum=,x,u;
     for(int i=;i<=n;i++)
         if(f[i][k]!=INF)q[++sum]=i,vis[i]=true;
     while(t!=sum){
         t++;if(t>=mod)t-=mod;x=q[t];
         for(int i=head[x];i;i=a[i].next){
             u=a[i].to;
             if(f[x][k]+a[i].dis<f[u][k]){
                 f[u][k]=f[x][k]+a[i].dis;
                 if(!vis[u]){
                     vis[u]=true;
                     sum++;if(sum>=mod)sum-=mod;q[sum]=u;
                 }
             }
         }
         vis[x]=false;
     }
 }
 int ans[<<];int c[][],ts[<<];
 bool check(int states){
     for(int i=;i<=;i++){
         if(!ts[i])continue;
         if((ts[i]&states) && ((states&ts[i])!=ts[i]))return false;
     }
     return true;
 }
 void work(){
     int x,y,z;
     memset(f,/,sizeof(f));INF=f[][];
     n=gi();m=gi();col=gi();
     for(int i=;i<=m;i++){
         x=gi();y=gi();z=gi();
         init(x,y,z);init(y,x,z);
     }
     for(int i=;i<=col;i++){
         x=gi();y=gi();
         f[y][<<(P++)]=;
         c[x][++c[x][]]=P-;
     }
     for(int i=;i<=;i++)
         for(int j=,tmp=c[i][];j<=tmp;j++)
             ts[i]|=(<<c[i][j]);
     tot=(<<P)-;
     for(int s=;s<=tot;s++){
         for(int i=;i<=n;i++)
             for(int j=(s-)&s;j;j=(j-)&s){
                 if(f[i][j]+f[i][s-j]<f[i][s]){
                     f[i][s]=f[i][j]+f[i][s-j];
                 }
             }
         spfa(s);
     }
     memset(ans,/,sizeof(ans));
     for(int s=;s<=tot;s++)
         for(int i=;i<=n;i++)
             if(f[i][s]<ans[s])ans[s]=f[i][s];
     for(int s=;s<=tot;s++){
         if(!check(s)){
             ans[s]=INF;
             continue;
         }
         for(int j=(s-)&s;j;j=(j-)&s){
             if(!check(j))continue;
             ans[s]=min(ans[s],ans[j]+ans[s-j]);
         }
     }
     printf("%d\n",ans[tot]);
 }
 int main()
 {
     freopen("pp.in","r",stdin);
     work();
     return ;
 }