[NOIP 2016 提高组] 愤怒的小鸟

GitHub跳转原题关系图返回列表

枚举两只小猪反推出一条合法下凹抛物线,把每条抛物线离散成一个覆盖集合,再做最少集合覆盖的状压 DP。

OJ: luogu

题目 ID: P2831

难度:提高+/省选-

标签:状态压缩动态规划几何最小覆盖

日期: 2026-06-21 05:36

题意

每次可以发射一只沿着 y = ax^2 + bx(且 a<0)飞行的小鸟。

如果轨迹经过某只小猪,这只小猪就会被消灭。 问最少需要多少只小鸟才能消灭所有小猪。

思路

先看一个小数据回溯:

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

const double EPS = 1e-8;

int T;
int n, m_dummy;
double x[20], y[20];
int cover[20][20];
int ans;

bool equal_double(double a, double b) {
    return fabs(a - b) < EPS;
}

void dfs(int mask, int used) {
    if (used >= ans) {
        return;
    }
    if (mask == (1 << n) - 1) {
        ans = min(ans, used);
        return;
    }

    int first = 0;
    while (mask & (1 << first)) {
        first++;
    }

    dfs(mask | (1 << first), used + 1);
    for (int j = 0; j < n; j++) {
        if (cover[first][j]) {
            dfs(mask | cover[first][j], used + 1);
        }
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    // brute.cpp:小数据回溯选下一条抛物线。
    cin >> T;
    while (T--) {
        cin >> n >> m_dummy;
        for (int i = 0; i < n; i++) {
            cin >> x[i] >> y[i];
        }

        for (int i = 0; i < n; i++) {
            cover[i][i] = 1 << i;
            for (int j = 0; j < n; j++) {
                if (i == j) {
                    continue;
                }
                cover[i][j] = 0;
                if (equal_double(x[i], x[j])) {
                    continue;
                }
                double a = (y[i] / x[i] - y[j] / x[j]) / (x[i] - x[j]);
                if (a >= -EPS) {
                    continue;
                }
                double b = y[i] / x[i] - a * x[i];
                int mask = 0;
                for (int k = 0; k < n; k++) {
                    double yy = a * x[k] * x[k] + b * x[k];
                    if (equal_double(yy, y[k])) {
                        mask |= 1 << k;
                    }
                }
                cover[i][j] = mask;
            }
        }

        ans = n;
        dfs(0, 0);
        cout << ans << '\n';
    }
    return 0;
}

正解的关键不是直接搜索抛物线,而是先把“有意义的抛物线”离散出来。

注意到一条过原点的二次函数 y=ax^2+bx,只要再给出两只小猪,就能唯一确定 a,b

所以可以枚举两只小猪 i,j

  1. 解出对应的 a,b
  2. a<0,说明这是一条合法轨迹
  3. 再检查这条轨迹还能经过哪些小猪

这样每条合法抛物线都会对应到一个覆盖集合 cover[i][j]

接下来就是状压 DP:

  • dp[mask] 表示消灭 mask 这些小猪最少需要多少只小鸟

DP 转移方程

cover[first][j] 表示一条经过 firstj 的合法抛物线能消灭的小猪集合,则:

dp[maskcover[first][j]]=min(dp[maskcover[first][j]], dp[mask]+1) dp[mask \mid cover[first][j]] =\min(dp[mask \mid cover[first][j]],\ dp[mask]+1)

如果只单独打一只 first,就把 cover[first][j] 换成 1<<first

转移时,找到第一只还没被打掉的猪 first

  • 可以单独打一只它
  • 或者枚举另一只猪 j,用 cover[first][j] 一次打一批

因为任意最优方案中的一条抛物线,只要能打一只以上的猪,就一定会在某个 cover[i][j] 中出现,所以不会漏解。

代码

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

const double EPS = 1e-8;

int T;
int n, m_dummy;
double x[20], y[20];
int cover[20][20];
int dp[1 << 18];

bool equal_double(double a, double b) {
    return fabs(a - b) < EPS;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    cin >> T;
    while (T--) {
        cin >> n >> m_dummy;
        for (int i = 0; i < n; i++) {
            cin >> x[i] >> y[i];
        }

        for (int i = 0; i < n; i++) {
            cover[i][i] = 1 << i;
            for (int j = 0; j < n; j++) {
                if (i == j) {
                    continue;
                }
                cover[i][j] = 0;
                if (equal_double(x[i], x[j])) {
                    continue;
                }

                double a = (y[i] / x[i] - y[j] / x[j]) / (x[i] - x[j]);
                if (a >= -EPS) {
                    continue;
                }
                double b = y[i] / x[i] - a * x[i];

                int mask = 0;
                for (int k = 0; k < n; k++) {
                    double yy = a * x[k] * x[k] + b * x[k];
                    if (equal_double(yy, y[k])) {
                        mask |= 1 << k;
                    }
                }
                cover[i][j] = mask;
            }
        }

        int full = 1 << n;
        for (int mask = 0; mask < full; mask++) {
            dp[mask] = n;
        }
        dp[0] = 0;

        for (int mask = 0; mask < full; mask++) {
            if (dp[mask] >= n) {
                continue;
            }
            int first = -1;
            for (int i = 0; i < n; i++) {
                if (!(mask & (1 << i))) {
                    first = i;
                    break;
                }
            }
            if (first == -1) {
                continue;
            }

            dp[mask | (1 << first)] = min(dp[mask | (1 << first)], dp[mask] + 1);
            for (int j = 0; j < n; j++) {
                if (cover[first][j]) {
                    dp[mask | cover[first][j]] = min(dp[mask | cover[first][j]], dp[mask] + 1);
                }
            }
        }

        cout << dp[full - 1] << '\n';
    }
    return 0;
}

复杂度

预处理 O(n3)O(n^3),状压 DP 约 O(n2n)O(n 2^n)

总结

这题的核心是把几何问题离散化成集合覆盖问题。 一旦把抛物线预处理成覆盖集合,后面的部分就是标准状压 DP。

一图流解析

这张图把本题的建模、关键转移、实现检查和训练方法压缩到一页,适合读完正文后复盘。

一图流解析