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;
}

 

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