[BZOJ 4774] 修路

题目描述：

给出 N 个点 和 M条边 以及 d
对于任意 1<=i<=d 要使得 i 与 n-i+1联通
求最小代价

题目分析：

emmmm
本题是最小斯坦纳树的题目.
具体什么是斯坦纳树呢，就是使得图上的某些特定点联通.
本题就是要我们求一个最小生成斯坦纳树.
我的理解就是图上的状压DP.
令f[i][s]表示以i为根节点且特定点的存在情况为状态最小联通代价
g[s]则为状态s的最小联通代价
如果求出了f[i][s]，那么求g[s]也就很容易了
对于f[i][s]的转移，可以有两种
1：是枚举子集取Min即可
2：是通过在状态相同时，我们可以通过最短路的转移方法进行求解f[i][s]
最后做一下子集DP求出g[s]即可

题目链接：

BZOJ 4774

Ac 代码：

#include <cstdio>
#include <iostream>
#include <queue>
#include <cstring>
const int maxm=20004; 
int inf;
int f[maxm][1<<8],g[1<<8],n,m,d;
int head[maxm<<1],to[maxm<<1],net[maxm<<1],cost[maxm<<1],cnt;
bool vis[maxm];
std::queue<int> dl;
void addedge(int u,int v,int c)
{
    ++cnt;
    to[cnt]=v,cost[cnt]=c,net[cnt]=head[u],head[u]=cnt;
}
void spfa(int sta)
{
    //while(!dl.empty()) dl.pop();
    for(int i=1;i<=n;i++)
     if(f[i][sta]<inf) dl.push(i),vis[i]=1;
    while(!dl.empty())
    {
        int x=dl.front();
        dl.pop();
        vis[x]=0;
        for(int i=head[x];i;i=net[i])
        if(f[to[i]][sta]>f[x][sta]+cost[i])
        {
            f[to[i]][sta]=f[x][sta]+cost[i];
            if(!vis[to[i]]) vis[to[i]]=1,dl.push(to[i]);
        }
    }
}
bool check(int sta)
{
    for(int i=0;i<d;i++)
    if((sta>>i)&1)
     if(!(sta>>(i+d)&1)) return 0;
    return 1;
}
int main()
{
    scanf("%d%d%d",&n,&m,&d);
    for(int i=1;i<=m;i++)
    {
        int u,v,c;
        scanf("%d%d%d",&u,&v,&c);
        addedge(u,v,c),addedge(v,u,c);
    }
    memset(f,127/3,sizeof(f)),memset(g,127/3,sizeof(g));
    inf=f[0][0];
    for(int i=1;i<=d;++i) f[i][1<<(i-1)]=0,f[n-i+1][1<<(d+i-1)]=0;
    for(int s=0;s<(1<<(d+d));s++)
    {
        for(int i=1;i<=n;i++)
         for(int j=(s-1)&s;j;j=(j-1)&s)
          f[i][s]=std::min(f[i][s],f[i][j]+f[i][s-j]);
        spfa(s);
        for(int i=1;i<=n;i++) g[s]=std::min(g[s],f[i][s]);
    }

    for(int s=0;s<(1<<(d+d));s++)
    {
        for(int i=(s-1)&s;i;i=(i-1)&s)
         if(check(i)&&check(s-i))
          g[s]=std::min(g[s],g[i]+g[s-i]);

    }
    int ans=g[(1<<(d+d))-1]==inf?-1:g[(1<<(d+d))-1];
    printf("%d\n",ans);
    return 0;
}