【NOIP2017】洛谷3959 宝藏题解（状压DP）

题目：luogu3959.
题目大意：给定一张$n$个点和$m$条边的图，让你选择一棵生成树，并选定一个根，那么这棵生成树的价格极即为每个节点的价格之和，一个节点的介个为这个节点到根的所经过的节点数乘上它到它的父亲的边权.现在要求输出最小价格.
$1\le n\le 12,1\le m\le 10^3$.

显然最小生成树是错误的，又看到这么小的数据范围，一看就知道是个状压DP.

设$f[i][S]$表示深度为$i$，当前连入生成树的点集为$S$的最小代价，然后我们发现并不能直接转移，怎么办？

考虑求出一个数组$g[S_1][S_2]$表示从状态$S_1$转移到状态$S_2$时，新加入的边的边权和，很容易用$O(4^n)$或$O(n^2{3}^n)$的时间复杂度处理出这个数组.

每次转移的时候，若从集合$S_1$转移到集合$S_2$，深度变化为$k$到$k+1$，就加上$g[S_1][S_2]\ast k$的贡献，这样做时间复杂度为$O(n{3}^n)$.

总空间复杂度为$O(4^n)$，时间复杂度为$O(3^n{n}^2)$.

代码如下：

#include<bits/stdc++.h>
using namespace std;

#define Abigail inline void
typedef long long LL;

const int N=12,INF=((1<<30)-1)/144;

int n,m,e[N+9][N+9];
int go[(1<<N)+9][(1<<N)+9];

void Get_go(){
  for (int g1=0;g1<1<<n;++g1){
  	int t=~g1&(1<<n)-1;
  	for (int g2=t;g2;g2=t&g2-1)
	  for (int i=0;i<n;++i){
	  	if (g1>>i&1^1) continue;
	    int t=INF;
	    for (int j=0;j<n;++j)
	      if (g2>>j&1) t=min(t,e[j+1][i+1]);
	    go[g2][g1|g2]+=t;
	  }
  }
}

int dp[N+9][(1<<N)+9];

void Get_dp(){
  for (int i=1;i<=n;++i)
    for (int g=0;g<1<<n;++g) dp[i][g]=INF;
  for (int i=0;i<n;++i) dp[1][1<<i]=0;
  for (int i=1;i<n;++i)
    for (int g1=0;g1<1<<n;++g1){
      int t=~g1&(1<<n)-1;
      for (int g2=t;g2;g2=t&g2-1)
        dp[i+1][g1|g2]=min(dp[i+1][g1|g2],dp[i][g1]+go[g1][g1|g2]*i);
    }
}

int ans;

Abigail into(){
  scanf("%d%d",&n,&m);
  for (int i=1;i<=n;++i)
    for (int j=1;j<=n;++j) e[i][j]=INF;
  for (int i=1;i<=m;++i){
  	int x,y,v;
  	scanf("%d%d%d",&x,&y,&v);
  	e[x][y]=e[y][x]=min(e[x][y],v);
  }
}

Abigail work(){
  Get_go();
  Get_dp();
  ans=INF;
  for (int i=1;i<=n;++i) ans=min(ans,dp[i][(1<<n)-1]);
}

Abigail outo(){
  printf("%d\n",ans);
}

int main(){
  into();
  work();
  outo();
  return 0;
}