题解 CF739E 【Gosha is hunting】

· · 题解

Solution

这其实是个wqs二分套wqs二分

我们要做的是用凸优化，把时间复杂度降到$\mathcal{O(n\log a\log b)}

显然这是一个凸函数，为什么呢？

因为你每次多用一个Poke Ball或者一个Ultra Ball能增加的捕捉到的Pokemons的期望一定比前一次的来得少 所以它的导数一定是递减的所以它是凸的 (不要以为递增函数一定不是凸函数，这是片面的理解)

所以我们可以在a这一维上进行wqs二分，也可以在b这一维上进行 每次dp转移的时候就只需要\mathcal{O(n)}的时间，每用一个Poke Ball减少mid的dp值，用Ultra Ball也是一样的

Warning

判断dp的a,b与我们现有的a,b比较时，要用>=不能用>

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define T 40
#define N 2010
using namespace std;
typedef double LD;
const LD INF = 1e5;

LD u[N], v[N], q[N], L, R;
int n, a, b;

struct Node{
    LD val;
    int a, b;
    Node(LD V = 0, int A = 0, int B = 0) {
        val = V; a = A; b = B;
    }
    inline bool operator < (const Node &o) const {
        return val < o.val;
    }
    inline Node operator + (const Node &o) const {
        return Node(val + o.val, a + o.a, b + o.b);
    }
}dp[N];

inline void check_agn(LD x, LD y) {
    dp[0] = Node();
    for (int i = 1; i <= n; ++i) {
        dp[i] = Node(-INF, 0, 0);
        dp[i] = max(max(dp[i - 1], dp[i - 1] + Node(q[i] - x - y, 1, 1)), max(dp[i - 1] + Node(u[i] - x, 1, 0), dp[i - 1] + Node(v[i] - y, 0, 1)));
    }
}

inline void check(LD v) {
    L = 0, R = 1;
    for (int i = 1; i <= T; ++i) {
        LD mid = (L + R) / 2;
        check_agn(v, mid);
        if (dp[n].b >= b) L = mid;
        else R = mid;
    }
    check_agn(v, L);
}

int main() {
    scanf("%d%d%d", &n, &a, &b);
    for (int i = 1; i <= n; ++i) {
        scanf("%lf", &u[i]);
    }
    for (int i = 1; i <= n; ++i) {
        scanf("%lf", &v[i]);
    }
    for (int i = 1; i <= n; ++i) {
        q[i] = u[i] + v[i] - u[i] * v[i];
    }
    LD l = 0, r = 1;
    for (int i = 1; i <= T; ++i) {
        LD mid = (l + r) / 2;
        check(mid);
        if (dp[n].a >= a) l = mid;
        else r = mid;
    }
    check(l);
//  printf("%.5Lf %.5Lf\n %d %d \n", l, L, dp[n].a, dp[n].b);
    printf("%.5lf\n", dp[n].val + l * a + L * b);
    return 0;
}