最大流——poj3308 (模板)

类似于二分匹配的构图,相乘可以用log 后相加解决

View Code
#include<iostream>
#include<cmath>
#include<stdio.h>
#include<string.h>
using namespace std;
const double inf=500.0;
const int MAX=105;
struct node
{
int v,next;
double c;
}g[100000];
int num[MAX],dis[MAX],cur[MAX],adj[MAX],pre[MAX];
int s,t,e,vn,r,c,l;

void init(int nn,int ss,int tt)
{
e=0;
memset(adj,-1,sizeof(adj));
s=ss;//源点
t=tt;//汇点
vn=nn;//vt表示总的定点数
}
void add(int u,int v,double c)
{
g[e].v=v; g[e].c=c; g[e].next=adj[u]; adj[u]=e++;
g[e].v=u; g[e].c=0; g[e].next=adj[v]; adj[v]=e++;
}
double sap()
{
int i,u,v,flag;
double aug=inf,flow=0.0;
for(i=0;i<=vn;i++)
{
dis[i]=num[i]=0;
cur[i]=adj[i];
}
u=s;
num[0]=vn;
while(dis[s]<vn)
{
flag=0;
for(i=cur[u];i!=-1;i=g[i].next)
{
v=g[i].v;
if(g[i].c>0.0&&dis[u]==dis[v]+1)
{
flag=1;
cur[u]=i;
pre[v]=u;
aug=min(aug,g[i].c);
u=v;
if(u==t)
{
flow+=aug;
//cout<<flow<<endl;
while(u!=s)
{
u=pre[u];
g[cur[u]].c-=aug;
g[cur[u]^1].c+=aug;
}
aug=inf;
}
break;
}
}
if(flag)
continue;
if(--num[dis[u]]==0)
return flow;
for(dis[u]=vn,i=adj[u];i!=-1;i=g[i].next)
{
v=g[i].v;
if(g[i].c>0.0&&dis[v]<dis[u])
{
dis[u]=dis[v];
cur[u]=i;
}
}
dis[u]++;
num[dis[u]]++;
if(u!=s)
u=pre[u];
}
return flow;
}
int main()
{
int i,j,test,a,b;
double cost;
scanf("%d",&test);
while(test--)
{
scanf("%d %d %d",&r,&c,&l);
init(r+c+2,0,r+c+1);
for(i=1;i<=r;i++)
{
scanf("%lf",&cost);
add(s,i,log(cost));
}
for(i=1;i<=c;i++)
{
scanf("%lf",&cost);
add(i+r,t,log(cost));
}
for(j=1;j<=l;j++)
{
scanf("%d %d",&a,&b);
add(a,r+b,inf);
}
// cout<<sap()<<endl;
printf("%.4lf\n",exp(sap()));
}
return 0;
}

ps:别人的模板,自己又加工了下,还是不错的

原文地址:https://www.cnblogs.com/huhuuu/p/2422118.html