题目描述
小铭铭最近进入了某情报部门,该部门正在被如何建立安全的通道连接困扰。
该部门有 n 个情报站,用 1 到 n 的整数编号。给出 m 对情报站 ui;vi 和费用 wi,表示情
报站 ui 和 vi 之间可以花费 wi 单位资源建立通道。
如果一个情报站经过若干个建立好的通道可以到达另外一个情报站,那么这两个情报站就
建立了通道连接。形式化地,若 ui 和 vi 建立了通道,那么它们建立了通道连接;若 ui 和 vi 均
与 ti 建立了通道连接,那么 ui 和 vi 也建立了通道连接。
现在在所有的情报站中,有 p 个重要情报站,其中每个情报站有一个特定的频道。小铭铭
面临的问题是,需要花费最少的资源,使得任意相同频道的情报站之间都建立通道连接。
n≤1000 m≤3000 p≤10
分析
最终目标是使同一组的特殊点位于一个联通块里。由于特殊点很少,可以考虑用斯坦纳树来做。
首先是斯坦纳树的一搬讨论,设状态f[i][s],然后子集DP+spfa。
然而不同组的特殊点不一定在一个联通块里,所以预处理f数组后,还要另一个子集DP来合并答案。设g[s],状态s中为1的所有联通块都已合并。
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=1005,M=6005,S=1025,inf=1e8;
typedef long long LL;
int n,m,tot,d,h[N],e[M],nxt[M],w[M],ans,D[N],f[N][S],g[S],s,ST[N],id[N],belong[N];
bool v[N];
char c;
int read()
{
for (c=getchar();c<'0' || c>'9';c=getchar());
int x=c-48;
for (c=getchar();c>='0' && c<='9';c=getchar()) x=x*10+c-48;
return x;
}
void add(int x,int y,int z)
{
e[++tot]=y; nxt[tot]=h[x]; w[tot]=z; h[x]=tot;
}
void init(int x,int y)
{
if (y>d)
{
for (int i=1;i<=n;i++) if (f[i][x]<g[x]) g[x]=f[i][x];
return;
}
init(x,y+1); if (ST[y]>0) init(x|ST[y],y+1);
}
int main()
{
n=read(); m=read(); d=read();
while (m--)
{
int x=read(),y=read(),w=read();
add(x,y,w); add(y,x,w);
}
memset(f,42,sizeof(f));
s=1<<d;
for (int i=0;i<d;i++)
{
belong[i]=read(); id[i]=read(); ST[id[i]]=1<<i;
f[id[i]][ST[id[i]]]=0;
}
for (int st=1;st<s;st++)
{
tot=0;
for (int i=1;i<=n;i++)
{
if (ST[i] && (st&ST[i])==0) continue;
for (int sub=(st-1)&st;sub;sub=(sub-1)&st)
{
int x=ST[i]|sub,y=ST[i]|(st^sub);
if (f[i][st]>f[i][x]+f[i][y]) f[i][st]=f[i][x]+f[i][y];
}
}
memset(v,0,sizeof(v));
for (int i=1;i<=n;i++) if (f[i][st]<inf)
{
v[i]=1; D[tot=(tot+1)%n]=i;
}
for (int i=0;i!=tot;)
{
int x=D[i=(i+1)%n];
for (int j=h[x];j;j=nxt[j]) if (f[e[j]][st|ST[e[j]]]>f[x][st]+w[j])
{
f[e[j]][st|ST[e[j]]]=f[x][st]+w[j];
if (!v[e[j]])
{
v[e[j]]=1; D[tot=(tot+1)%n]=e[j];
}
}
v[x]=0;
}
}
memset(g,42,sizeof(g));
for (int i=1;i<=d;i++)
{
ST[i]=0;
for (int j=0;j<d;j++) if (belong[j]==i) ST[i]|=(1<<j);
}
init(0,1);
for (int j=1;j<s;j++)
{
for (int sub=(j-1)&j;sub;sub=(sub-1)&j) g[j]=min(g[j],g[sub]+g[j-sub]);
}
printf("%d\n",g[s-1]);
return 0;
}