codeforces 739E Gosha is hunting

你要抓神奇宝贝！ 现在一共有

N

N 只神奇宝贝。 你有

a

a 个『宝贝球』和

b

b 个『超级球』。 『宝贝球』抓到第

i

i 只神奇宝贝的概率是

p_i

pi​ ，『超级球』抓到的概率则是

u_i

ui​ 。 不能往同一只神奇宝贝上使用超过一个同种的『球』，但是可以往同一只上既使用『宝贝球』又使用『超级球』（都抓到算一个）。 请合理分配每个球抓谁，使得你抓到神奇宝贝的总个数期望最大，并输出这个值。

设dp[i][j][k] 前i个神奇宝贝 用了j个a球 k个b球 期望的最大值是多少 那么直接

dp即可 但是显然复杂度是不对的n^3 那么考虑wqs二分 发现函数单增 那么我们二分一下每次使用a,b两种球会带来的减少量 复杂度n*log*log即可解决

#include<bits/stdc++.h>
#define eps 1e-8
using namespace std;
const int N=2200;
int n,numa[N],numb[N],A,B;double a[N],b[N],dp[N];
inline void gao(double ca,double cb){
    for (int i=1;i<=n;++i){
        dp[i]=dp[i-1];numa[i]=numa[i-1];numb[i]=numb[i-1];
        if(dp[i-1]+a[i]-ca>dp[i]){
            dp[i]=dp[i-1]+a[i]-ca;numa[i]=numa[i-1]+1;numb[i]=numb[i-1];
        }
        if (dp[i-1]+b[i]-cb>dp[i]){
            dp[i]=dp[i-1]+b[i]-cb;numa[i]=numa[i-1];numb[i]=numb[i-1]+1;
        }
        if (dp[i-1]+a[i]+b[i]-a[i]*b[i]-ca-cb>dp[i]){
            dp[i]=dp[i-1]+a[i]+b[i]-a[i]*b[i]-ca-cb;numa[i]=numa[i-1]+1;numb[i]=numb[i-1]+1;
        }
    }
}
int main(){
    freopen("cf.in","r",stdin);
    scanf("%d%d%d",&n,&A,&B);
    for (int i=1;i<=n;++i) scanf("%lf",&a[i]);
    for (int i=1;i<=n;++i) scanf("%lf",&b[i]);
    double l1=0,r1=1,l2,r2;
    while(l1<r1-eps){
        double mid=(l1+r1)/2;
        l2=0;r2=1;
        while(l2<r2-eps){
            double mid1=(l2+r2)/2;
            gao(mid,mid1);
            if (numb[n]<=B) {r2=mid1;if (numb[n]==B) break;}else l2=mid1; 
        }
        if (numa[n]<=A) {r1=mid;if (numa[n]==A) break;}else l1=mid;
    }printf("%f\n",dp[n]+A*r1+B*r2);
    return 0;
}