problem
参与考古挖掘的小明得到了一份藏宝图,藏宝图上标出了 n n n 个深埋在地下的宝藏屋, 也给出了这 n n n 个宝藏屋之间可供开发的 m m m 条道路和它们的长度。
小明决心亲自前往挖掘所有宝藏屋中的宝藏。但是,每个宝藏屋距离地面都很远,也就是说,从地面打通一条到某个宝藏屋的道路是很困难的,而开发宝藏屋之间的道路则相对容易很多。
小明的决心感动了考古挖掘的赞助商,赞助商决定免费赞助他打通一条从地面到某个宝藏屋的通道,通往哪个宝藏屋则由小明来决定。
在此基础上,小明还需要考虑如何开凿宝藏屋之间的道路。已经开凿出的道路可以任意通行不消耗代价。每开凿出一条新道路,小明就会与考古队一起挖掘出由该条道路所能到达的宝藏屋的宝藏。另外,小明不想开发无用道路,即两个已经被挖掘过的宝藏屋之间的道路无需再开发。
新开发一条道路的代价是:
L × K \mathrm{L} \times \mathrm{K} L×K
L \mathrm{L} L 代表这条道路的长度, K \mathrm{K} K 代表从赞助商帮你打通的宝藏屋到这条道路起点的宝藏屋所经过的宝藏屋的数量(包括赞助商帮你打通的宝藏屋和这条道路起点的宝藏屋)。
请你编写程序为小明选定由赞助商打通的宝藏屋和之后开凿的道路,使得工程总代价最小,并输出这个最小值。
数据范围: 1 ≤ n ≤ 12 1≤n≤12 1≤n≤12, 0 ≤ m ≤ 1000 0 \le m \le 1000 0≤m≤1000, v ≤ 500000 v \le 500000 v≤500000。
solution
下面的内容来源于 Henry__Huang 的博客。
设 f [ i ] [ j ] f[i][j] f[i][j] 表示当前树高为 i i i,已经选了的点集的集合为 j j j,那么状态转移方程为:
f [ i ] [ j ] = min k ∈ j { f [ i − 1 ] [ j x o r k ] + d i s [ j x o r k ] [ k ] ⋅ ( i − 1 ) } f[i][j]=\min_{k \in j}\{f[i-1][j \;\mathrm{xor} \;k]+dis[j \;\mathrm{xor} \; k][k]⋅(i-1)\} f[i][j]=k∈jmin{f[i−1][jxork]+dis[jxork][k]⋅(i−1)}
其中 d i s [ i ] [ j ] dis[i][j] dis[i][j] 表示从 i i i 这个已选点集加上下一层将要选的 j j j 这个点集所需要的最小花费。
这个转移方程应该挺显然的,就是在原树上加一些点构成新树。
d i s [ i ] [ j ] dis[i][j] dis[i][j] 递推式如下:
d i s [ i ] [ j ] = d i s [ i ] [ j x o r l o w b i t ( j ) ] + min k ∈ i { d [ log 2 l o w b i t ( j ) + 1 ] [ k ] } dis[i][j]=dis[i][j \;\mathrm{xor} \; \mathrm{lowbit}(j)]+\min_{k\in i}\{d[\log_2\mathrm{lowbit}(j)+1][k]\} dis[i][j]=dis[i][jxorlowbit(j)]+k∈imin{d[log2lowbit(j)+1][k]}
这就相当于是把 l o w b i t ( j ) \mathrm{lowbit}(j) lowbit(j) 的那个点先拿出来,最后再单独统计贡献。
时间复杂度 O ( n 2 n ) O(n2^n) O(n2n)。
小 tip:枚举 S S S 子集的方法。
for(int i=S;i;i=(i-1)&S)
code
#include<cstdio>
#include<cstring>
#include<algorithm>
#define lowbit(x) (x&(-x))
using namespace std;
const int N=15,S=(1<<12)+5;
int n,m,e[N][N],f[N][S],Q[S],dis[S][S],sum[S],Log[S];
int main() {
int x,y,z;
scanf("%d%d",&n,&m);
memset(e,0x3f,sizeof(e));
for(int i=1;i<=m;++i){
scanf("%d%d%d",&x,&y,&z);
e[x][y]=e[y][x]=min(e[x][y],z);
}
int sta=(1<<n)-1;
for(int i=2;i<=sta;++i) Log[i]=Log[i/2]+1;
for(int i=1;i<=sta;++i){
int cnt=0,Min;
for(int j=sta^i;j;j=(j-1)&(sta^i)) Q[++cnt]=j;
for(int j=cnt;j>=1;--j) {
dis[i][Q[j]]=dis[i][Q[j]^lowbit(Q[j])],Min=500000;
for(int k=1;k<=n;++k) if((1<<(k-1))&i) Min=min(Min,e[k][Log[lowbit(Q[j])]+1]);
dis[i][Q[j]]+=Min;
}
}
memset(f,0x3f,sizeof(f));
for(int i=0;i<n;++i) f[1][1<<i]=0;
for(int i=1;i<=sta;++i){
int x=i;
while(x) sum[i]++,x>>=1;
}
for(int i=2;i<=n;++i){
for(int j=(1<<i)-1;j<=sta;++j){
if(sum[j]<i) continue;
for(int k=j;k;k=(k-1)&j) f[i][j]=min(f[i][j],f[i-1][j^k]+dis[j^k][k]*(i-1));
}
}
int ans=1<<30;
for(int i=1;i<=n;++i) ans=min(ans,f[i][sta]);
printf("%d\n",ans);
return 0;
}