bzoj4006 管道连接【斯坦纳树+子集dp】

解题思路：

题目要求的即为最小斯坦纳森林。
我们可以枚举颜色种类的一个子集S，把颜色属于S的关键点提出来求一次最小斯坦纳树，设最小值为g[S]，那么最后子集dp一波：g[S]=min(g[S],g[i]+g[S^i])(i$\in$S)。

求最小斯坦纳树的方法源于状压dp：
设f[i][s]表示以i为根的子树中关键点联通状态为s的边权和最小值，那么转移方程有两种，分别为：
f[i][s]=min(f[i][s],f[i][j]+f[i][s^j])(j$\in$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;
}