CF2041C Cube

注意到 n ≤ 12 n \le 12 n12,于是乎考虑使用状压 DP

根据常见的套路,考虑将三维中的两维状压。

不妨设 d p i , s t 1 , s t 2 dp_{i,st1,st2} dpi,st1,st2 表示前 i i i 层已经选了恰好 i i i 个点(从题面可以看出每一层仅可能选 1 1 1 个点),这 i i i 个点占用 y y y 轴方向的状态为 s t 1 st1 st1,占用 z z z 轴方向的状态为 s t 2 st2 st2


首先有一种朴素的思路:直接枚举 i i i上一层 s t 1 st1 st1 s t 2 st2 st2,然后枚举这一层选择的点。

设目前枚举的第 i i i 层选择的点为 ( i , x , y ) (i,x,y) (i,x,y),状压之后为 w x = 2 x wx=2^x wx=2x w y = 2 y wy=2^y wy=2y,默认 x , y x,y x,y 0 0 0 开始。

转移方程很好写:

d p i , s t 1 ∣ w x , s t 2 ∣ w y = min ⁡ ( d p i , s t 1 ∣ w x , s t 2 ∣ w y , d p i − 1 , s t 1 , s t 2 + a i , x , y ) dp_{i,st1|wx,st2|wy}=\min(dp_{i,st1|wx,st2|wy},dp_{i-1,st1,st2}+a_{i,x,y}) dpi,st1∣wx,st2∣wy=min(dpi,st1∣wx,st2∣wy,dpi1,st1,st2+ai,x,y)

枚举 i , x , y i,x,y i,x,y 需要 O ( n 3 ) O(n^3) O(n3),然后 s t 1 st1 st1 s t 2 st2 st2 又是 O ( 2 2 n ) O(2^{2n}) O(22n)。合起来是过不去的。

需要优化。


我们考虑优化枚举范围:发现 s t 1 st1 st1 实际上取不到那么多, s t 2 st2 st2 也是。具体算一下,在 i i i 固定的情况下,可能的种数正好是 C n i C_{n}^i Cni

于是考虑将 i i i 对应的所有情况都存起来,到时候直接取即可。

for (int i = 1; i < (1 << n); i++)//枚举所有的状态
		v[__builtin_popcount(i) - 1].push_back(i);//存起来

__builtin_popcount() 是数一个数二进制下的 1 1 1 的位数。

于是我们发现 O ( n 3 × 2 2 n ) O(n^3 \times 2^{2n}) O(n3×22n) 在这种实现方法下是跑不满的,因为 C n i C_{n}^i Cni 的和总共也就 2 n 2^n 2n

#include <bits/stdc++.h>
using namespace std;
int n;
const int N = 12;
int dp[N][(1 << N) + 5][(1 << N) + 5];//注意空间
vector<int> v[N];
int a[N][N][N];

int main() {
	cin >> n;
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
			for (int k = 0; k < n; k++)
				cin >> a[i][j][k];
	for (int i = 1; i < (1 << n); i++)
		v[__builtin_popcount(i) - 1].push_back(i);
	for (int i = 0; i < n; i++)
		for (int j = 0; j < (1 << n); j++)
			for (int k = 0; k < (1 << n); k++)
				dp[i][j][k] = 1e9;
	for (int i = 0; i < n; i++) {
		if (i == 0) {//注意特判
			for (int j = 0, wj = 1; j < n; j++, wj *= 2)
				for (int k = 0, wk = 1; k < n; k++, wk *= 2)
					dp[i][wj][wk] = min(a[i][j][k], dp[i][wj][wk]);
		} else {
			for (auto p1 : v[i - 1])
				for (int j = 0, wj = 1; j < n; j++, wj *= 2) {
					if (p1 & wj)
						continue;
					for (auto p2 : v[i - 1])
						for (int k = 0, wk = 1; k < n; k++, wk *= 2) {
							if (p2 & wk)
								continue;
							dp[i][p1 | wj][p2 | wk] = min(dp[i - 1][p1][p2] + a[i][j][k], dp[i][p1 | wj][p2 | wk]);
						}
				}
		}
	}
	cout << dp[n - 1][(1 << n) - 1][(1 << n) - 1];
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值