题意:
小明有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;
}