[ 斯坦纳树 ] BZOJ4774

题解

#include<bits/stdc++.h>
using namespace std;
const int INF=1e9;
const int N=10010;
const int D=8;
int h[N],nx[N<<1],t[N<<1],w[N<<1],num;
queue<int>q;
int k,n,m,d,x,y;
int f[N][1<<D];
int Ans[1<<D];
bool b[N];
inline void Add(int x,int y,int z){
    t[++num]=y;w[num]=z;nx[num]=h[x];h[x]=num;
}
inline void Spfa(int y){
    while(!q.empty()){
        int x=q.front();q.pop();
        b[x]=0;
        for(int i=h[x];i;i=nx[i])
            if(f[t[i]][y]>f[x][y]+w[i]){
                f[t[i]][y]=f[x][y]+w[i];
                if(!b[t[i]])q.push(t[i]),b[t[i]]=1;
            }
    }
}
inline void Init(){
    memset(f,63,sizeof(f));memset(Ans,63,sizeof(Ans));
    for(int i=1;i<=d;i++)f[i][1<<i-1]=f[n-i+1][1<<i+d-1]=0;
    for(int i=1;i<(1<<d+d);i++){
        memset(b,0,sizeof(b));
        for(int j=1;j<=n;j++)
                for(int k=(i-1)&i;k;k=(k-1)&i)
                    f[j][i]=min(f[j][i],f[j][k]+f[j][i^k]);
        for(int j=1;j<=n;j++)
            if(f[j][i]<INF)q.push(j),b[j]=1;
        Spfa(i);
        for(int j=1;j<=n;j++)Ans[i]=min(Ans[i],f[j][i]);
    }
}
inline bool Check(int x){
    for(int i=0;i<d;i++)
        if((x>>i)&1&&!((x>>i+d)&1))return 0;
    return 1;
}
int main(){
    scanf("%d%d%d",&n,&m,&d);
    while(m--)scanf("%d%d%d",&x,&y,&k),Add(x,y,k),Add(y,x,k);
    Init();
    for(int i=1;i<(1<<d+d);i++)
        for(int j=(i-1)&i;j;j=(j-1)&i)
            if(Check(j)&&Check(i^j))
                Ans[i]=min(Ans[i],Ans[j]+Ans[i^j]);
    if(Ans[(1<<d+d)-1]>INF)puts("-1");else cout<<Ans[(1<<d+d)-1]<<endl;
    return 0;
}