【Codeforces 739E】 Gosha is hunting【费用流】

题意:

    小明有a个神奇宝贝球，b个超级神奇宝贝球，现在有n只神奇宝贝，A球B球分别有p1i,p2i的概率抓到i神奇宝贝,一个神奇宝贝被一种球只能抓一次,问抓到神奇宝贝个数的期望是多少

题解:

    看官方题解看不太懂，无意间看到了wkc的代码，一下子看懂了

    费用流可做

    A=n+1,B=n+2

    vs向A连(a,0)

    vs向B连(b,0)

    A向i连(1,p1i)

    B向i连(1,p2i)

    i向vt连(1,0)

    i向vt连(1,-p1*p2)

    最后一条边就代表了扔了两个神奇宝贝球的期望

#include<cmath>
#include<stack>
#include<queue>
#include<bitset>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define PB push_back
#define MP make_pair
#define ll long long
#define MS(a,b) memset(a,b,sizeof(a))
#define LL (rt<<1)
#define RR (rt<<1|1)
#define lson l,mid,LL
#define rson mid+1,r,RR
#define pii pair<int,int>
#define pll pair<ll,ll>
#define lb(x) (x&(-x))
void In(){freopen("in.in","r",stdin);}
void Out(){freopen("out.out","w",stdout);}
const int N=1e5+10;
const int M=3e5+10;
const int Mbit=1e6+10;
const int inf=0x3f3f3f3f;
const ll mod=1e9+7;
const double eps=1e-8;
struct Edge
{
    int u,v,cap,next;
    double cost;
}edge[M<<2];
int NE;
int head[N],pp[N],Q[N];
double dist[N];
bool vis[N];
void init()
{
    NE=0;
    memset(head,-1,sizeof(head));
}
void add(int u,int v,int cap,double cost)
{
    edge[NE].u=u;edge[NE].v=v;edge[NE].cap=cap;edge[NE].cost=cost;
    edge[NE].next=head[u];head[u]=NE++;

    edge[NE].u=v;edge[NE].v=u;edge[NE].cap=0;edge[NE].cost=-cost;
    edge[NE].next=head[v];head[v]=NE++;
}
bool SPFA(int s,int t,int n)
{
    int i,u,v,l=0,r=1;
    memset(vis,0,sizeof(vis));
    memset(pp,-1,sizeof(pp));
    for(i=0;i<n;i++) dist[i]=-inf;
    vis[s]=1;dist[s]=0;
    Q[0]=s;
    while(l!=r)
    {
        u=Q[l++];vis[u]=0;
        if(l==N)l=0;
        for(i=head[u];i!=-1;i=edge[i].next)
        {
            v=edge[i].v;
            if(edge[i].cap&&dist[v]<dist[u]+edge[i].cost-eps)
            {
                dist[v]=dist[u]+edge[i].cost;
                pp[v]=i;
                if(!vis[v])
                {
                    Q[r++]=v;
                    vis[v]=1;
                    if(r==N)r=0;
                }
            }
        }
    }
    return dist[t]>-inf;
}
pair<int,double> MCMF(int s,int t,int n)
{
    int flow=0,i,minflow;
    double mincost=0;
    while(SPFA(s,t,n))
    {
        minflow=inf;
        for(i=pp[t];i!=-1;i=pp[edge[i].u])
            minflow=min(edge[i].cap,minflow);
        flow+=minflow;
        for(i=pp[t];i!=-1;i=pp[edge[i].u])
            edge[i].cap-=minflow,edge[i^1].cap+=minflow;
        mincost+=dist[t]*minflow;
    }
    return make_pair(flow,mincost);
}
double p1[N],p2[N];
int main()
{
    //In();
    int n,a,b;
    while(~scanf("%d%d%d",&n,&a,&b)){
        init();
        int A=n+1,B=n+2,vs=0,vt=B+1,NV=vt+1;
        add(vs,A,a,0);
        add(vs,B,b,0);
        for(int i=1;i<=n;i++){
            scanf("%lf",&p1[i]);
            add(A,i,1,p1[i]);
        }
        for(int i=1;i<=n;i++){
            scanf("%lf",&p2[i]);
            add(B,i,1,p2[i]);
        }
        for(int i=1;i<=n;i++){
            add(i,vt,1,0);
            add(i,vt,1,-p1[i]*p2[i]);
        }
        printf("%.6f\n",MCMF(vs,vt,NV).second);
    }
    return 0;
}