Peach Blossom Spring HDU - 4085 （斯坦纳树）

最近补南昌邀请赛题发现A 是 斯坦纳树，然后先去学习了一下。这道好像是11年北京区域赛题。斯坦纳树其实是状压dp的一种，借助spfa来进行同类型之间的转移，整个状态转移的顺序是最精髓的地方。
状态 dp[i][sta] 表示以 i 为根，联通点的状态为 sta 的最小代价。
有两种转移：

1. dp[i][sta] = dp[i][s] + dp[i][s^sta] (其中s是sta的子集）
2. dp[v][sta] = dp[u][sta] + w[u][v] (同种状态的转移，可以用最短路来求）

关于转移的顺序，从第一个转移方程可以看出，要从sta 小的向大的集合转移。所以要先求出小的集合。然后对于每一种sta，由转移方程1转移后，有些点的该状态下的答案已经被更新，将这些点入队，在该sta的情况下去跑最短路，更新那些没有办法由子集转移到sta的点。
然后重复这个过程。

大致题意

有n个村庄，通过m条路相连，要求将前k个点，和后k个点一对一联通，最少需要修理多少长度的路径。（没有规定这个k对点的联通方式，只要一对一进行，也就是有k*k种）。

思路

先对原图做一遍斯坦纳树，然后发现题目要求的实际是一个斯坦纳树林的合并。至于为什么不能直接用min(f[i][s])；因为f[i][s]的最小值不一定是由合法的状态更新而来的，合法状态就符合题意的配对方式的状态，所以最终还要取出合法状态进行一遍转移。这个check函数便是判断状态是否合法的，只要前k和后k个数相同即可算作合法，被我注释的代码是理解错配对的方式了。

代码

#include<bits/stdc++.h>
using namespace std;
#define maxn 55
#define maxm 1005
#define ll long long int
#define INF 0x3f3f3f3f
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
#define mem(a) memset(a,0,sizeof(a))
#define sqr(x) (x*x)
#define inf (ll)2e18+1
#define PI acos(-1)
#define ls x<<1
#define rs x<<1|1
ll read(){
    ll x=0,f=1ll;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
     while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
     return f*x;
}
int T,n,m,k;
struct node{int v,w,nxt;}ed[maxm<<1];
int head[maxn],tot;
void add(int x,int y,int z){ed[++tot]={y,z,head[x]};head[x]=tot; }
int f[maxn][1<<12],dp[1<<12];
queue<int>q;
bool inq[maxn];
void spfa(int sta){
    while(!q.empty()){
        int u=q.front();q.pop();inq[u]=0;
        for(int i=head[u];i;i=ed[i].nxt){
            int v=ed[i].v;
            if(f[v][sta]>f[u][sta]+ed[i].w){
                f[v][sta]=f[u][sta]+ed[i].w;
                if(inq[v]==0){
                    inq[v]=1;
                    q.push(v);
                }
            }
        }
    }
}
/*bool check(int sta){
    inc(i,0,k-1)if(((1<<i)&sta)^((1<<(k+i))&sta))return false;
    return true;
}*/
bool check(int sta){
    int cnt = 0;
    for(int i = 0;i<k;i++) {
        if(sta & (1<<i)) cnt++;
        if(sta & (1<<(k+i))) cnt--;
    }
    return cnt == 0;
}
int main()
{
    T=read();
    while(T--){
        n=read();m=read();k=read();
        int s=(1<<(2*k))-1;
        tot=0;inc(i,0,n)head[i]=0;
        inc(i,1,n)inc(j,0,s)f[i][j]=INF;
        inc(i,1,k)f[i][1<<(i-1)]=0,f[n-i+1][1<<(k+i-1)]=0;
        int x,y,z;
        inc(i,1,m){
            x=read();y=read();z=read();
            add(x,y,z);add(y,x,z);
        }
        inc(i,0,n)inq[i]=0;
        inc(sta,0,s){
            inc(i,1,n){
                for(int j=sta;j;j=(j-1)&sta){
                    f[i][sta]=min(f[i][sta],f[i][j]+f[i][sta^j]);
                }
                if(f[i][sta]<INF){
                    inq[i]=1;
                    q.push(i);
                }
            }
            spfa(sta);
        }
        inc(i,0,s){
            dp[i]=INF;
            inc(j,1,n)dp[i]=min(dp[i],f[j][i]);
        }
        inc(sta,0,s){
            if(!check(sta))continue;
            for(int j=sta;j;j=(j-1)&sta){
                if(!check(j))continue;
                dp[sta]=min(dp[sta],dp[j]+dp[sta^j]);
            }
        }
        if(dp[s]<INF)printf("%d\n",dp[s]);
        else printf("No solution\n");
    }
    return 0;
}