注意到 n ≤ 12 n \le 12 n≤12,于是乎考虑使用状压 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,dpi−1,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;
}