4006: [JLOI2015]管道连接 斯坦纳树+状压dp

好题qwq。。
首先做一遍裸的斯坦纳树。（似乎我写的姿势不太对？好慢啊qwq）。
然后我们可以发现，答案要求的并不一定是一棵斯坦纳树，也可能是一个斯坦纳树森林，对于这个我们可以再次DP一下，思路与刚才类似。
令dp[i]表示至少包含状态为i时点的斯坦纳树的最小费用，为了保证这个状态是合法的，我们要保证有i时要包含频道为i的所有点，假设这个合法的状态为k。那么dp[i]=min(dp[i],f[j][k]),1<=j<=n。这样求出来依然是一棵树。。然后可以用刚才类似的思想，即把状态“分离”，dp[i]=min(dp[i],dp[j]+dp[i-j]),这样求出的最优答案可能就是森林了。
（据说不考虑森林裸打斯坦纳树可以90分卧槽？）
感觉get了状压的新姿势。
最后还是想说，好题qwq。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define inf 0x3f3f3f3f
#define M 1005
using namespace std;
int n,m,p,t,w,cnt;
struct node {int c,d;} a[11];
int f[1005][1024];
int head[1005],list[6005],next[6005],key[6005];
bool v[1005];
int q[1005],dp[1024];
int ww[11];
inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
inline bool cmp(node a,node b)
{
    return a.c<b.c;
}
inline void insert(int x,int y,int z)
{
    next[++cnt]=head[x];
    head[x]=cnt;
    list[cnt]=y;
    key[cnt]=z;
}
inline void spfa(int sta)
{
    for (int i=1;i<=n;i++) w=(w+1)%M,q[w]=i,v[i]=1;
    while (t!=w)
    {
        t=(t+1)%M;
        int x=q[t];
        for (int i=head[x];i;i=next[i])
            if (f[list[i]][sta]>f[x][sta]+key[i])
            {
                f[list[i]][sta]=f[x][sta]+key[i];
                if (!v[list[i]])
                {
                    w=(w+1)%M;
                    v[list[i]]=1;
                    q[w]=list[i];
                }
            }
        v[x]=0;
    }
}
int main()
{
    n=read(); m=read(); p=read(); t=0; w=1;
    for (int i=1;i<=m;i++)
    {
        int u=read(),v=read(),w=read();
        insert(u,v,w); insert(v,u,w);
    }
    memset(f,0x3f,sizeof(f));
    for (int i=1;i<=p;i++)
    {
        int c=read(),d=read();
        f[d][1<<i-1]=0;
        ww[c]|=(1<<i-1); 
    }
    for (int sta=1;sta<1<<p;sta++)
    {
        for (int i=1;i<=n;i++)
            for (int s=sta&(sta-1);s;s=sta&(s-1))
                f[i][sta]=min(f[i][sta],f[i][s]+f[i][sta-s]);
        spfa(sta);
    }
    for (int i=1;i<1<<p;i++) 
    {
        int k=0; dp[i]=1e9;
        for (int j=1;j<=p;j++) if (i&(1<<j-1)) k|=ww[j];  
        for (int j=1;j<=n;j++) dp[i]=min(dp[i],f[j][k]);
        for (int j=(i-1)&i;j;j=(j-1)&i) dp[i]=min(dp[i],dp[j]+dp[i-j]);
    }
    cout << dp[(1<<p)-1] << endl;
    return 0;
}