题目:愤怒的小鸟
看了某些dalao的题解才会的 O ( T ⋅ 2 N ⋅ N ) O(T \cdot 2^N \cdot N) O(T⋅2N⋅N),自己只想出来了 O ( T ⋅ 2 N ⋅ N 2 ) O(T \cdot 2^N \cdot N^2) O(T⋅2N⋅N2)。
这道题 N ≤ 18 N \le 18 N≤18 所以可以开状压, d p i dp_i dpi 记录已经被打掉的猪的编号的二进制为 i i i 时的最优解,例如 d p 5 dp_5 dp5 代表打掉第 0 , 2 0,2 0,2 头猪时的最优解; h i t i j hit_{ij} hitij 代表打掉第 i , j i,j i,j 头时能打掉的所有猪的二进制(因为打掉要第 i , j i,j i,j 头所以抛物线可以确定)
可以优化的一点就是每次动规的起始位置,记录每种情况下第一个为 0 0 0 的二进制位。因为打的顺序不分先后,所以我们可以规定从编号小的往大的打,这样就可以通过存储每种情况下要打的编号来减少时间复杂度。
时间复杂度 O ( T ⋅ 2 N ⋅ N ) O(T\cdot 2^N\cdot N) O(T⋅2N⋅N),空间复杂度 O ( 2 N ) O(2^N) O(2N)。
#include <cstring>
#include <cstdio>
#include <cmath>
const int MAXN = 20;
const double eps = 1e-6;
struct Pos {
double x, y;
} p[MAXN];
int hit[MAXN][MAXN], begin[1 << MAXN], dp[1 << MAXN], n;
inline double pow2(double n) { return n * n; }
// 已知两点坐标,解 y=ax^2+bx
inline void solve(double &a, double &b, Pos p1, Pos p2) {
a = ((p1.x / p2.x) * p2.y - p1.y) / (p1.x * (p2.x - p1.x));
b = (p1.y - pow2(p1.x) * a) / p1.x;
}
inline int min(int a, int b) {
return a < b ? a : b;
}
int main() {
for (int i = 1; i < (1 << 18); i++) {
for (; begin[i] < 18 && (i & (1 << begin[i])); begin[i]++);
}
int t, m; scanf("%d", &t);
while (t--) {
memset(hit, 0, sizeof(hit)); // 初始化
memset(dp, 0x7f, sizeof(dp));
dp[0] = 0;
scanf("%d%d", &n, &m); // 输入
for (int i = 0; i < n; i++) {
scanf("%lf%lf", &p[i].x, &p[i].y);
}
for (int i = 0; i < n; i++) {
for (int m = 0; m < n; m++) {
// 如果x坐标差值过小则无法在同一抛物线上(也可能是同一点)
if (fabs(p[i].x - p[m].x) < eps) continue;
double a, b; solve(a, b, p[i], p[m]); // 求解
// 即判断a是否小于0(等于不行),这里需要取到-eps防止卡精度
if (a > -eps) continue;
for (int k = 0; k < n; k++) { // 枚举每一点看在不在抛物线上
if (fabs(a * pow2(p[k].x) + b * p[k].x - p[k].y) < eps) {
hit[i][m] |= (1 << k);
}
}
}
}
for (int i = 0; i < (1 << n); i++) { // 状压dp
int beg = begin[i]; // beg是必须要打的
dp[i | (1 << beg)] = min(dp[i | (1 << beg)], dp[i] + 1);
for (int m = 0; m < n; m++) { // 枚举每种情况
dp[i | hit[beg][m]] = min(dp[i | hit[beg][m]], dp[i] + 1);
}
}
// (1<<n)-1对应的二进制是11111...(共n个1)
printf("%d\n", dp[(1 << n) - 1]);
}
return 0;
}