CF739E Gosha is hunting（期望dp+wqs二分）

首先我们有朴素dp，f[i][a][b]表示对于前i个小精灵用a个a球，b个b球可以获得的最大期望值。
枚举第i个小精灵的方式转移即可。$O(n^3)$
考虑优化这个dp，我们发现这个问题好像还可以费用流…诶，他好像是凸的…诶…wqs二分！
好像两维都可以wqs二分呢！orz
于是$O(nlog^2n)$解决了这个问题。
你问我为啥是凸的…我哪会啊qaq
注意精度问题，因为你最后还要*A,所以精度要1e-8

#include <bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define N 2010
#define eps 1e-8
int n,A,B,numa[N],numb[N];
double a[N],b[N],f[N];
inline int solve(double ca,double cb,bool fl){
    for(int i=1;i<=n;++i){
        f[i]=f[i-1];numa[i]=numa[i-1];numb[i]=numb[i-1];
        if(f[i-1]+a[i]-ca>f[i]){
            f[i]=f[i-1]+a[i]-ca;numa[i]=numa[i-1]+1;numb[i]=numb[i-1];
        }if(f[i-1]+b[i]-cb>f[i]){
            f[i]=f[i-1]+b[i]-cb;numa[i]=numa[i-1];numb[i]=numb[i-1]+1;
        }if(f[i-1]+a[i]+b[i]-a[i]*b[i]-ca-cb>f[i]){
            f[i]=f[i-1]+a[i]+b[i]-a[i]*b[i]-ca-cb;
            numa[i]=numa[i-1]+1;numb[i]=numb[i-1]+1;
        }
    }return fl?numa[n]:numb[n];
}
int main(){
//  freopen("a.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(r1-l1>eps){
        double mid1=(r1+l1)/2;
        l2=0,r2=1;
        while(r2-l2>eps){
            double mid2=(r2+l2)/2;
            if(solve(mid1,mid2,0)>B) l2=mid2;else r2=mid2;
        }if(solve(mid1,r2,1)>A) l1=mid1;else r1=mid1;
    }solve(r1,r2,0);printf("%lf\n",f[n]+B*r2+A*r1);
    return 0;
}