POJ - 3436 ACM Computer Factory(最大流+拆点+输出流量路径)

链接:https://cn.vjudge.net/problem/POJ-3436

题意:N个制作电脑的工厂,一个完整的电脑有P个部件。每个工厂每小时都可以把若干个半成品电脑变成若干半成品或成品电脑。也就是给他若干个有某些零件的半成品电脑(有些必须有,有些必须没有,有些可有可无),这个工厂会制作出若干个有某些零件的电脑。问这N个工厂一小时能生产多少完整的电脑,并且输出工厂之间的联系。(形如u、v、w,代表从u运输给v(或从v运输给u)w个它自己生产的电脑产品。)

思路:我们将工厂拆成两个点,因为工厂有输入、有输出,给原料,才能生产啊。

1、将不需要任何零件的工厂(i)与源点(s)连边,容量为该工厂的生产量。

2、工厂拆为的两点之间(i与i+n)连边,容量为该工厂的生产量。

3、将符合输出(u+n)、输入(v)的工厂之间连边,容量为inf。

4、将生产成品电脑的工厂(i+n)与汇点(t)连边,容量为i的产量。

至于输出流量路径,求完最大流后,每条边反向弧中的流量便是这条边流过的流量。

#include <cstdio>
#include <iostream>
#include <cstring>
#include <queue>
#include <algorithm>
#define ll long long
using namespace std;
const int N = 55;
const int M = 1e4+10;
const int inf = 0x3f3f3f3f;
struct node
{
	int s[N],d[N],q;
}a[N];
struct edge
{
	int from,to,ca,nxt;
	edge(){}
	edge(int f,int t,int c,int nx)
	{
		from=f,to=t,ca=c,nxt=nx;
	}
}g[M],ans[M];
int head[N<<1],cnt;
int deep[N<<1],mp[N<<1][N<<1],pre[N<<1],cur[N<<1];
int p,n,s,t;
void Init()
{
	s=0;
	t=n<<1|1;
	cnt=0;
	for(int i=s;i<=t;i++)
		head[i]=-1;	
	for(int i=1;i<=n;i++)
		for(int j=i+1;j<=n;j++)
			mp[i][j]=mp[j][i]=0;
	//memset(head,-1,sizeof(head));
	//memset(mp,0,sizeof(mp));
}
void add(int u,int v,int w)
{
	g[cnt]=edge(u,v,w,head[u]),head[u]=cnt++;
}
bool judge0(int b[])
{
	for(int i=1;i<=p;i++)
		if(b[i]==1)
			return 0;
	return 1;
}
bool judge(int a[],int b [])
{
	for(int i=1;i<=p;i++)
		if(a[i]!=b[i]&&b[i]!=2)
			return 0;
	return 1;
}
bool judge1(int b[])
{
	for(int i=1;i<=p;i++)
		if(b[i]!=1)
			return 0;
	return 1;
}
bool bfs()
{
	memset(deep,0,sizeof(deep));
	queue<int> q;
	int u,v;
	deep[s]=1;
	q.push(s);
	while(!q.empty())
	{
		u=q.front();
		q.pop();
		if(u==t) return 1;
		for(int i=head[u];~i;i=g[i].nxt)
		{
			v=g[i].to;
			if(!deep[v]&&g[i].ca>0)
			{
				deep[v]=deep[u]+1;
				q.push(v);
			}
		}
	}
	return deep[t]!=0;
}
int getid(int x)
{
	return x>n?x-n:x;
}
int dfs(int u,int flow)
{
	if(u==t||!flow) return flow;
	int v,nowflow,ans=0;
	for(int &i=cur[u];~i;i=g[i].nxt)
	{
		v=g[i].to;
		if(deep[v]==deep[u]+1&&g[i].ca>0)
		{
			nowflow=dfs(v,min(flow,g[i].ca));
			if(nowflow)
			{
				ans+=nowflow;
				flow-=nowflow;
				//mp[getid(u)][getid(v)]+=nowflow;
				g[i].ca-=nowflow;
				g[i^1].ca+=nowflow;
				if(!flow) break;
			} 	
		}	
	} 
	if(!ans) deep[u]=0;
	return ans;
}
void dinic()
{
	int maxflow=0,flow,now;
	while(bfs())
	{
		for(int i=s;i<=t;i++)
			cur[i]=head[i];
		while(flow=dfs(s,inf))
			maxflow+=flow;
	}
	if(!maxflow)
	{
		puts("0 0");
		return ;
	}
	int tot=0,u,v;
	for(u=1+n;u<=(n<<1);u++)
		for(int i=head[u];~i;i=g[i].nxt)
			if(g[i].to<=n&&g[i^1].ca>0&&u!=g[i].to+n)
				ans[++tot]=edge(u-n,g[i].to,g[i^1].ca,0);							
	printf("%d %d\n",maxflow,tot);			
	for(int i=1;i<=tot;i++)
		printf("%d %d %d\n",ans[i].from,ans[i].to,ans[i].ca);	
}
int main(void)
{
	while(~scanf("%d%d",&p,&n))
	{
		Init();
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i].q);
			for(int j=1;j<=p;j++)
				scanf("%d",&a[i].s[j]);
			for(int j=1;j<=p;j++)
				scanf("%d",&a[i].d[j]);				
		}
		for(int i=1;i<=n;i++)
			if(judge0(a[i].s))
			{
				add(s,i,inf);
				add(i,s,0);
			}
		for(int i=1;i<=n;i++)
		{
			add(i,i+n,a[i].q);
			add(i+n,i,0);
		}
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
			{
				if(i==j) continue;
				if(judge(a[i].d,a[j].s))
				{
					add(i+n,j,inf);
					add(j,i+n,0);
				}
			}			
		for(int i=1;i<=n;i++)
			if(judge1(a[i].d))
			{
				add(i+n,t,inf);
				add(t,i+n,0);	
			}
		dinic();	
	}
	return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章