Codeforces739E Gosha is hunting

题意:现在有n个精灵,两种精灵球各m1和m2个,每个精灵单独使用第一种精灵球有pi的概率被捕获,单独使用第二种精灵球有ui的概率被捕获,同时使用有1-(1-pi)*(1-ui)的概率被捕获.一种精灵球在一个精灵身上只能用一次,但你可以在一个精灵上用两种精灵球.求最优策略下期望获得精灵的只数.

如果一只精灵上不能同时用两种精灵球,那么就是一个显然的费用流建图,点A表示第一种精灵球,点B表示第二种精灵球,源点向A,B各连一条流量等于对应精灵球数目的边(费用为0),A,B分别向每个精灵连一条流量为1,费用为对应的pi/ui的边,每个精灵连出一条流量为1的边指向汇点,限制只能用一个精灵球,跑最大费用流即可(这里最大费用流等于最大费用最大流,因为精灵球用得越多期望抓捕的精灵数至少不会减少).

现在我们需要考虑同时使用两种精灵球的情况,但并不能简单地把精灵连向汇点的边流量+1,因为这样相加的时候我们重复计算了两种精灵球都生效的情况,这个概率是pi*ui.

那么我们想法减去多出的一部分即可,同时还要保证只使用一个精灵球的时候不减去这部分,那么我们从每个精灵向汇点引出两条流量为1的边,一条费用为0,一条费用为-pi*ui,使用第一个精灵球时最长路一定走费用为0的边,使用第二个精灵球时必须走第二条边从而去除了重复的情况.跑最大费用流即可.

坑点:这道题的费用是实数,spfa比较大小的时候一定要用eps,不然会卡出TLE….

#include<cstdio>
#include<cstring>
const int maxn=2005,maxm=20005;
struct edge{
  int to,next,w;double cost;
}lst[maxm];int len=0,first[maxn];
void addedge(int a,int b,int w,double cost){
  lst[len].to=b;lst[len].next=first[a];lst[len].w=w;lst[len].cost=cost;first[a]=len++;
  lst[len].to=a;lst[len].next=first[b];lst[len].w=0;lst[len].cost=-cost;first[b]=len++;
}
bool inq[maxn];
int s,t,T,head,tail,q[maxn],vis[maxn],prt[maxn];double dis[maxn];
bool spfa(){
  head=tail=0;q[tail++]=s;inq[s]=true;
  vis[s]=++T;dis[s]=0;prt[s]=-1;
  while(head!=tail){
    int x=q[head++];head%=maxn;inq[x]=false;
    for(int pt=first[x];pt!=-1;pt=lst[pt].next){
      if(lst[pt].w==0)continue;
      if(vis[lst[pt].to]!=T||dis[x]+lst[pt].cost-dis[lst[pt].to]>1e-8){//这个地方神坑…直接比较大小TLE到死…
    vis[lst[pt].to]=T;dis[lst[pt].to]=dis[x]+lst[pt].cost;
    prt[lst[pt].to]=pt;
    if(!inq[lst[pt].to]){
      inq[lst[pt].to]=true;q[tail++]=lst[pt].to;tail%=maxn;
    }
      }
    }
  }
  return vis[t]==T;
}
double maxcost(){
  double ans=0;
  while(spfa()&&dis[t]>0){
    ans+=dis[t];
    for(int pt=prt[t];pt!=-1;pt=prt[lst[pt^1].to]){
      lst[pt].w--;lst[pt^1].w++;
    }
  }
  return ans;
}
double p[maxn],u[maxn];
int main(){
  memset(first,-1,sizeof(first));
  int n,m1,m2;scanf("%d%d%d",&n,&m1,&m2);
  s=0;t=n+3;
  addedge(s,n+1,m1,0);addedge(s,n+2,m2,0);
  for(int i=1;i<=n;++i)scanf("%lf",p+i);
  for(int i=1;i<=n;++i)scanf("%lf",u+i);
  for(int i=1;i<=n;++i){
    addedge(n+1,i,1,p[i]);addedge(n+2,i,1,u[i]);
    addedge(i,t,1,0);addedge(i,t,1,-p[i]*u[i]);
  }
  printf("%.5f
",maxcost());
  return 0;
}