题目大意
给定n个点,给出m条边。
给出k,选择使用一些边,求满足
∑ki=1(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]);
}