[poj] 3686 The Windy's || 最小费用最大流

原题

有N个工件要在M个机器上加工，有一个NM的矩阵描述其加工时间。
同一时间内每个机器只能加工一个工件，问加工完所有工件后，使得平均加工时间最小（等待的时间+加工的时间）。
也就是说倒数第k个加工的对总时间的贡献是ka[i][j]，拆点为k个，建费用为k*a[i][j]，容量为1的边即可。

#include<cstdio>
#include<queue>
#include<cstring>
#define N 1010*1010
#define inf 0x3f3f3f3f
using namespace std;
int n,m,head[N],dis[N],cur[N],ans,cnt=2,s,t,ANS,T;
queue <int> q;
bool vis[N];
struct hhh
{
    int to,next,w,cost;
}edge[1000005];

int read()
{
    int ans=0,fu=1;
    char j=getchar();
    for (;(j<'0' || j>'9') && j!='-';j=getchar()) ;
    if (j=='-') fu=-1,j=getchar();
    for (;j>='0' && j<='9';j=getchar()) ans*=10,ans+=j-'0';
    return ans*fu;
}

void add(int u,int v,int w,int c)
{
    edge[cnt].to=v;
    edge[cnt].w=w;
    edge[cnt].next=head[u];
    edge[cnt].cost=c;
    head[u]=cnt++;
}

void addEdge(int u,int v,int w,int c)
{
    add(u,v,w,c);
    add(v,u,0,-c);
}

bool bfs()
{
    for (int i=s;i<=t;i++)
	vis[i]=0,cur[i]=head[i],dis[i]=inf;
    q.push(s);
    dis[s]=0;
    vis[s]=1;
    while(!q.empty())
    {
	int r=q.front();
	q.pop();
	vis[r]=0;
	for (int i=head[r],v;i;i=edge[i].next)
	{
	    v=edge[i].to;
	    if (edge[i].w>0 && dis[r]+edge[i].cost<dis[v])
	    {
		dis[v]=dis[r]+edge[i].cost;
		if (!vis[v])
		{
		    vis[v]=1;
		    q.push(v);
		}
	    }
	}
    }
    return dis[t]!=inf;
}

int dfs(int x,int f)
{
    if (x==t) return ANS+=f*dis[t],f;
    int ha=0,now;
    vis[x]=1;
    for (int &i=cur[x],v;i;i=edge[i].next)
    {
	v=edge[i].to;
	if (vis[v]) continue;
	if (edge[i].w>0 && dis[v]==dis[x]+edge[i].cost)
	{
	    now=dfs(v,min(f-ha,edge[i].w));
	    if (now)
	    {
		ha+=now;
		edge[i].w-=now;
		edge[i^1].w+=now;
	    }
	}
	if (ha==f) return ha;
    }
    return ha;
}

void init()
{
    memset(head,0,sizeof(head));
    cnt=2;
    ANS=0;
    ans=0;
}

int main()
{
    T=read();
    while (T--)
    {
	init();
	n=read();
	m=read();
	s=0;
	t=n+m*n+1;
	for (int i=1;i<=n;i++)
	    addEdge(s,i,1,0);
	for (int i=1,a;i<=n;i++)
	    for (int j=1;j<=m;j++)
	    {
		a=read();
		for (int k=1;k<=n;k++)
		addEdge(i,j*n+k,1,k*a);
	    }
	for (int i=1;i<=m;i++)
	    for (int j=1;j<=n;j++)
		addEdge(i*n+j,t,1,0);
	while (bfs()) ans+=dfs(s,inf);
	printf("%f
",ANS*1.0/n);
    }
    return 0;
}