POJ2584 T-Shirt Gumbo 最大流

算是最大流基础题了吧。

构图：多出一个源点start，一个汇点final

源点到5个衣服的边权值是每个衣服的数量数。

然后每个衣服到匹配的人的边权值都是1，然后每个人到汇点的边权值都是1.

这样最大流一下，其实就是用最大流求最大匹配。

#include<iostream>
#include<queue>
using namespace std;
const int N=200;
const int M=2000;
char range[N][2];
bool visit[N];
int pre[N];
int k=1;
int start;
int final;
int n;
char getcha[6]={'0','S','M','L','X','T'};
int getin(char a)
{
	switch(a)
	{
	case 'S':return 1;
	case 'M':return 2;
	case 'L':return 3;
	case 'X':return 4;
	case 'T':return 5;
	}
}
struct node
{
	int v,w,next,re;
}edge[M];
int edgehead[N];
void addedge(int a ,int b,int w)
{
	edge[k].v=b;
	edge[k].w=w;
	edge[k].next=edgehead[a];
	edge[k].re=k+1;
	edgehead[a]=k++;
	edge[k].v=a;
	edge[k].w=0;
	edge[k].next=edgehead[b];
	edge[k].re=k-1;
	edgehead[b]=k++;
}
bool bfs()
{
	memset(visit,0,sizeof(visit));
	memset(pre,0,sizeof(pre));
	queue<int> que;
	que.push(start);
	visit[start]=true;
	while(!que.empty())
	{
		int now=que.front();
		que.pop();
		for(int i=edgehead[now];i!=0;i=edge[i].next)
		{
			if(edge[i].w>0&&!visit[edge[i].v])
			{
				visit[edge[i].v]=true;
				que.push(edge[i].v);
				pre[edge[i].v]=i;
				if(edge[i].v==final)
					return true;
			}
		}
	}
	return false;
}
void solve()
{
	int ans=0;
	while(bfs())
	{
		int now=final;
		while(now!=start)
		{
			int t=pre[now];
			edge[t].w-=1;
			edge[edge[t].re].w+=1;
			now=edge[edge[t].re].v;
		}
		ans++;
	}
	if(ans<n)
		printf("I'd rather not wear a shirt anyway...\n");
	else
		printf("T-shirts rock!\n");


}
int main()
{
	char str[20];
	while(scanf("%s %d",str,&n),str[0]=='S')
	{
		k=1;
		start=n+6;
		final=n+7;
		memset(edgehead,0,sizeof(edgehead));
		memset(edge,0,sizeof(edge));
		memset(pre,0,sizeof(pre));
		for(int i=1;i<=n;i++)
		{
			scanf("%s",range[i]);
			for(int j=getin(range[i][0]);j<=getin(range[i][1]);j++)
			{
				addedge(j,i+5,1);
			}
		}
		int tmp;
		for(int i=1;i<=5;i++)
		{
			scanf("%d",&tmp);
			addedge(n+5+1,i,tmp);
		}
		for(int i=1;i<=n;i++)
		{
			addedge(5+i,n+5+2,1);
		}
		solve();
		scanf("%s",str);
	}
	return 0;
}