[NOIP2017 提升组] 宝藏

[NOIP2017 提升组] 宝藏


状态压缩好题。下面咱们来一一分析一下这道题的整体想法。ios


容易发现,\(n\leq12\)暗示这道题时间复杂度为指数级别的。c++

因为题目代表测试

两个已经被挖掘过的宝藏屋之间的道路无需再开发。优化

因此,咱们开发的路径构成的只能是一棵树,不能是有环的图。spa

所以,咱们能够将题目抽象成:给定一张有\(n\)个点的图,求出一棵生成树(根节点任意)使 \(\sum_{u\in G,fa_u\in G} dep_u\times w(u,fa_u)\)最小。code

因为\(n\)很小,不难想到可使用状态压缩。ip

我最开始的想法是:枚举根节点\(root\),并定义以当前的根为根节点的状态:\(d(i,j,S)\)表明当前叶子结点\(j\)\(root\)距离为\(i\)时,正处理的点集\(S\)的最小值。开发

那么\(i\geq2\)时有以下转移:\(dp(i,j,S)=min(dp(i-1,k,S\setminus\{i\})+i\times d(i,k))\)get

\(j=1\)有:\(dp(i,j,S)=min(dp(x,k,S\setminus\{i\})+d(j,root))\)string

初始化:让\(dp(0,root,0)=0\),其他\(\infty\)

这样的作法效率大约是:\(O(n^42^{n})\),实际测试中全部的状态是跑不满的,故这种作法是能够经过本题目的时间限制的。


咱们注意到刚刚那个转移是至关混乱的,而且它会重复计算状态值。

咱们上述作法本质上是利用叶子结点向上逆推推的过程。那么,咱们是否能够反过来想——从一个结点向它的子结点扩展,考虑剩下的子结点集合的代价。

根据此,咱们来从新定义以前的状态表示:

\(dp(i,j,S)\)表明距离根节点距离\(i\)考虑到的当前结点为\(j\),从\(j\)开始“挖”的点集为\(S\)的最小代价和。换句话来说,咱们只考虑\(j\)为根的子树的代价。

那么有:\(dp(i,j,S)=min(dp(i+1,k,S'\setminus\{k\})+dp(i,j,S\setminus{S'})+d(j,k))\),这里面的\(S'\)表明的是\(S\)中的子集,而k是S‘中的元素。

不难观察到,方程相似于树形DP中子树合并这一重要思想。事实上,咱们利用它解决过树形依赖关系的一类问题(背包)。

初始化,咱们让\(dp(i,j,0)=0\),其他正无穷,最终答案为\(min(dp(0,i,U\setminus{i}))\)。时间复杂度为\(O(n^32^n)\),具体计算过于复杂,在此不展开。

在实现的过程当中,为优化常数,咱们能够预处理出全部状态下的元素个数、最小元素下标,以及将全部的点的下标平移。

另外,要注意边界条件。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#define RE register
#define CLR(x, y) memset(x,y,sizeof x)
#define FOR(i, x, y) for(RE int i=x;i<=y;++i)
#define ROF(i, x, y) for(RE int i=x;i>=y;--i)
using namespace std;
const int N = 13, S = 1 << N, INF = 1e9 + 5;
typedef long long LL;
template <class T> 
void read(T &x)
{
	bool mark = false;
	char ch = getchar();
	for(; ch < '0' || ch > '9'; ch = getchar()) if(ch == '-') mark = true;
	for(x = 0; ch >= '0' && ch <= '9'; ch = getchar()) x = (x << 3) + (x << 1) + ch - '0';
	if(mark) x = -x;
	return;
}
int n, m, t, siz[S] = {}, tt[S] = {};
LL ans = INF, d[N][N], dp[N][N][S];
int main()
{
	read(n), read(m);
	t = 1 << n, -- t;
	FOR(i, 0, n) 
	{
		d[i][i] = 0;
		FOR(j, i + 1, n) d[i][j] = d[j][i] = INF;
	}
	LL u, v, w;
	FOR(i, 1, m)
	{
		read(u), read(v), read(w);
		-- u,  -- v;
		d[u][v] = d[v][u] = min(d[u][v], w);
	}	
	FOR(i, 1, t) siz[i] = siz[i & (i - 1)] + 1;
	for(int i = 0; i < n; ++ i) tt[1 << i] = i;
	FOR(i, 1, t) tt[i] = tt[i & (-i)];
	CLR(dp, 0x3f);
	for(int i = 0; i < n; ++ i)
		for(int j = 0; j < n; ++ j)
			dp[i][j][0] = 0;
	for(int i = n - 2; i >= 0; -- i)
	{
		for(int j = 0; j < n; ++ j)
		{
			FOR(s, 1, t)
			{
				if((s & (1 << j)) || siz[t ^ s] <= i) continue;
				for(int S1 = s; S1; S1 = (S1 - 1) & s)
				{
					for(int tmp = S1, k = tt[S1]; tmp; k = tt[tmp ^= tmp & (-tmp)])
						if(d[j][k] < INF) dp[i][j][s] = min(dp[i][j][s], dp[i + 1][k][S1 ^ (1 << k)] + dp[i][j][s ^ S1] + (i + 1) * d[j][k]);
				}
			}
		}
	}
	for(int i = 0; i < n; ++ i) ans = min(ans, dp[0][i][t ^ (1 << i)]);
	printf("%lld\n", ans);
	return 0;  
}

总结:

  1. 当数据规模很小的时候,能够考虑状态压缩
  2. 在考虑树上问题时每每从根节点开始考虑(像倍增一类的题目是从本身点开始考虑);
相关文章
相关标签/搜索