解题思路:
题目要求的即为最小斯坦纳森林。
我们可以枚举颜色种类的一个子集S,把颜色属于S的关键点提出来求一次最小斯坦纳树,设最小值为g[S],那么最后子集dp一波:g[S]=min(g[S],g[i]+g[S^i])(i
∈
S)。
求最小斯坦纳树的方法源于状压dp:
设f[i][s]表示以i为根的子树中关键点联通状态为s的边权和最小值,那么转移方程有两种,分别为:
f[i][s]=min(f[i][s],f[i][j]+f[i][s^j])(j
∈
s),这个用子集dp先转移。
f[i][s]=min(f[i][s],f[j][s]+w(i,j),这个用SPFA后转移。
#include<bits/stdc++.h>
using namespace std;
int getint()
{
int i=0,f=1;char c;
for(c=getchar();(c<'0'||c>'9')&&c!='-';c=getchar());
if(c=='-')f=-1,c=getchar();
for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
return i*f;
}
const int N=1005,M=6005,INF=0x3f3f3f3f;
int n,m,p;
int tot,first[N],nxt[M],to[M],w[M];
int f[N][1<<10],g[1<<10];
bool exist[N];
queue<int>q;
struct node
{
int x,id;
inline friend bool operator < (const node &a,const node &b)
{return a.x<b.x;}
}a[15];
void add(int x,int y,int z)
{
nxt[++tot]=first[x],first[x]=tot,to[tot]=y,w[tot]=z;
}
int solve(int num)
{
for(int k=1;k<(1<<num);k++)
{
for(int i=1;i<=n;i++)
{
for(int j=(k-1)&k;j;j=(j-1)&k)
f[i][k]=min(f[i][k],f[i][j]+f[i][k^j]);
if(f[i][k]<INF)q.push(i),exist[i]=true;
}
while(!q.empty())
{
int u=q.front();
q.pop(),exist[u]=false;
for(int e=first[u];e;e=nxt[e])
{
int v=to[e];
if(f[v][k]>f[u][k]+w[e])
{
f[v][k]=f[u][k]+w[e];
if(!exist[v])q.push(v),exist[v]=true;
}
}
}
}
int res=INF;
for(int i=1;i<=n;i++)res=min(res,f[i][(1<<num)-1]);
return res;
}
int main()
{
//freopen("lx.in","r",stdin);
int x,y,z;
n=getint(),m=getint(),p=getint();
while(m--)
{
x=getint(),y=getint(),z=getint();
add(x,y,z),add(y,x,z);
}
for(int i=1;i<=p;i++)a[i].x=getint(),a[i].id=getint();
sort(a+1,a+p+1);
int last=-1,cnt=0;
for(int i=1;i<=p;i++)
{
if(a[i].x!=last)last=a[i].x,++cnt;
a[i].x=cnt;
}
for(int now=1;now<(1<<cnt);now++)
{
int tmp=0;
for(int i=1;i<=p;i++)
if(now&(1<<a[i].x-1))tmp++;
for(int i=1;i<=n;i++)
for(int j=1;j<(1<<tmp);j++)f[i][j]=INF;
tmp=0;
for(int i=1;i<=p;i++)
if(now&(1<<a[i].x-1))
f[a[i].id][1<<tmp++]=0;
g[now]=solve(tmp);
for(int i=(now-1)&now;i;i=(i-1)&now)
g[now]=min(g[now],g[i]+g[now^i]);
}
cout<<g[(1<<cnt)-1]<<'\n';
return 0;
}