感谢JYZ大佬,教会我了这道题的思想。
这题就是一道典型的斯坦纳树问题,求目标联通状态的最小权值和。
在斯坦纳树中,我们用到了状态压缩的思想,用一个二进制状态来表示当前的图中目标点是否在当前联通状态中。
我们定义f[i][opt]为当前在第i号节点,连通状态为opt的最小权值和。
我们可以分两种情况讨论:
对于第一种情况,我们可以用DP来处理;对于第二种情况,我们可以用spfa来更新。1.我们定义sub为当前opt的子集,则f[i][opt]=min(f[i][opt],f[i][sub]+f[i][opt^sub])
2.若第j号节点和第i号节点联通,则f[j][opt]=min(f[j][opt],f[i][opt]+dis[i][j])
最后是统计答案:
1.若当前的opt状态中,第i号节点和第n-i+1号节点都在或都不在这个联通状态中,则ans[opt]=min(ans[opt],f[i][opt])
2.定义sub为当前opt的子集,则ans[opt]=min(ans[opt],ans[sub]+ans[opt^sub])
最后,若目标连通状态的ans值为定义的无穷大,则输出-1,否则输出ans的值。
需要注意的是枚举opt的子集sub的方法。原理是什么,我也不太懂……
附上AC代码:
#include <cstdio>
#include <cctype>
#include <cstring>
#include <queue>
#define N 10010
using namespace std;
struct side{
int to,w,nt;
}s[N<<1];
queue <int> que;
int n,m,d,x,y,w,num,h[N],f[N][260],INF,sum,ans[260];
bool b[N];
inline char nc(){
static char ch[100010],*p1=ch,*p2=ch;
return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}
inline void read(int &a){
static char c=nc();int f=1;
for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
for (a=0;isdigit(c);a=a*10+c-'0',c=nc());
a*=f;return;
}
inline void add(int x,int y,int w){
s[++num]=(side){y,w,h[x]},h[x]=num;
s[++num]=(side){x,w,h[y]},h[y]=num;
}
inline void calc(){
memset(f,0x3f,sizeof f),INF=f[0][0],num=0;
for (int i=1; i<=d; ++i) f[i][1<<num++]=0;
for (int i=n-d+1; i<=n; ++i) f[i][1<<num++]=0;
sum=(1<<num)-1;
}
inline int min(int a,int b){
return a<b?a:b;
}
inline void spfa(int opt){
while (!que.empty()){
int p=que.front();que.pop(),b[p]=0;
for (int i=h[p]; i; i=s[i].nt)
if (f[s[i].to][opt]>f[p][opt]+s[i].w){
f[s[i].to][opt]=f[p][opt]+s[i].w;
if (!b[s[i].to]) que.push(s[i].to),b[s[i].to]=1;
}
}
}
inline void work(){
for (int opt=0; opt<=sum; ++opt){
for (int i=1; i<=n; ++i){
for (int sub=opt; sub; sub=(sub-1)&opt)
f[i][opt]=min(f[i][opt],f[i][sub]+f[i][opt^sub]);
if (f[i][opt]!=INF) que.push(i),b[i]=1;
}
spfa(opt);
}
}
inline bool check(int opt){
for (int i=0,j=(d<<1)-1; i<d; ++i,--j)
if (((opt&(1<<i))==0)!=((opt&(1<<j))==0)) return 0;
return 1;
}
int main(void){
read(n),read(m),read(d);
for (int i=1; i<=m; ++i) read(x),read(y),read(w),add(x,y,w);
calc(),work(),memset(ans,0x3f,sizeof ans);
for (int opt=0; opt<=sum; ++opt)
if (check(opt))
for (int i=1; i<=n; ++i)
ans[opt]=min(ans[opt],f[i][opt]);
for (int opt=0; opt<=sum; ++opt)
for (int sub=opt; sub; sub=(sub-1)&opt)
ans[opt]=min(ans[opt],ans[sub]+ans[opt^sub]);
printf("%d",ans[sum]==INF?-1:ans[sum]);
return 0;
}