旅行

题目大意

给定n个点，给出m条边。
给出k，选择使用一些边，求满足$\sum_{i=1}^k(i和n-i+1连通)$的最小边权和。

数据范围n,m<=10000,k<=4

斯坦纳树

模型：一个图中求给定的k个点相互连通的斯坦纳生成树
这题其实就是裸题。
如何求斯坦纳生成树呢
设d[i][j]表示以i为根的子树，k个关键点的选取状态为j(2进制数，0表示该点还没连通)时的最小权值和。
根据状态表示，我们可得转移方程：
d[i][j] = min(d[i1][j1]+w[i][i1]+d[i][j-j1])，其中j1|j= j,w[i][k]表示i，i1两点的距离
注意到这个方程在转移时可能会出现循环，因为我们不知道应该是用d[i][j]来更新d[i1][j]还是反过来。也就是说当j相等时，我们应该用最短路的思想来进行转移,我用的是spfa，当然也能用dij。
我们分两个部分更新，先O(n)处理d[i][j]=min(d[i][j1]+d[i][j-j1])
枚举j1有一个技巧：for (int j1=(j-1)&j;j1>0;j1=(j1-1)&j)
可以思考一下这为什么是对的。
再用spfa处理d[i][j]=min(d[i1][j]+w[i][i1])。
本题我们既然得出了d[i][j],设g[j]=min(d[i][j]),删除不合法的j(即不满足i和n-i+1连通)的g[j]的值删去。做个dp就行了。

代码

#include<cstdio>
#include<algorithm>
#include<cstring>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=10000+5,maxk=1<<8;
int d[maxn][maxk],G[maxk],f[maxn],g[maxn*2],next[maxn*2],c[maxn*2],st[maxn],q[maxn*4],bz[maxn];
int i,j,n,m,k,head,tail,w,num,l;
void add(int x,int y,int z){
    next[++num]=f[x];
    f[x]=num;g[num]=y;c[num]=z;
}
void read(int &n){
    char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    n=0;while (ch>='0'&&ch<='9') n=n*10+ch-'0',ch=getchar();    
}
void spfa(int s){
    head=0;
    while (head<tail){
        int x=q[++head],i=f[x];bz[x]=0;
        while (i>0){
            int y=g[i],z=st[y]|s;
            if (d[y][z]>d[x][s]+c[i]){
                d[y][z]=d[x][s]+c[i];
                if (z==s&&(bz[y]==0)) q[++tail]=y,bz[y]=1;
            }
            i=next[i];
        } 
    } 
}
bool pc(int s){
    fo(l,0,k/2-1){
        int x=1<<l,y=1<<(k-l-1);
        if (((x&s)>0&&(y&s)==0)||((x&s)==0&&(y&s)>0)) return 0;
    }return 1; 
} 
int main(){
    read(n);read(m);read(k);
    fo(i,1,m){
        int x,y,z;read(x),read(y),read(z);
        add(x,y,z);add(y,x,z);
    } 
    fo(i,1,k) st[i]=1<<i-1,st[n-i+1]=1<<(2*k-i);
    k+=k;num=0;
    memset(d,63,sizeof(d));w=d[0][0];
    fo(i,1,n) d[i][st[i]]=0;memset(G,63,sizeof(G));
    fo(j,1,(1<<k)-1){
        tail=0;
        fo(i,1,n){
            if(st[i]&&(st[i]&j)==0) continue; 
            for (int j1=(j-1)&j;j1>0;j1=(j1-1)&j){
                int x=st[i]|j1,y=st[i]|(j-j1);
                d[i][j]=min(d[i][j],d[i][x]+d[i][y]);
            } 
            if (d[i][j]<w)q[++tail]=i,bz[i]=1;

        }
        spfa(j);
        if (pc(j)) fo(i,1,n) G[j]=min(G[j],d[i][j]);
    }
    fo(j,1,(1<<k)-1) 
    for (int j1=(j-1)&j;j1>0;j1=(j1-1)&j) G[j]=min(G[j],G[j1]+G[j-j1]);
    if (G[(1<<k)-1]==w) G[(1<<k)-1]=-1;
    printf("%d",G[(1<<k)-1]);
}