Gym 101981I Magic Portion 最大流

• Magic Portion
  • 貪心是一定做不出來的，需要網絡流。
  • 我們建圖的方法十分簡單——
  • 源點與所有英雄點連一條流量爲1的邊，表示一個英雄只能殺初始只能殺一隻怪獸；
  • 所有英雄與其所能殺的所有怪獸分別連一條流量爲1的邊；
  • 所有怪獸與匯點連一條流量爲1的邊，表示一個怪獸只能死一次。
  • 下面考慮如何處理藥的問題：
  • 我們指定一個藥點，從源點向藥點連一條流量爲藥的數量的邊；
  • 從藥點向每個英雄連一條流量爲1的邊，表示每個英雄只能服1粒藥。
  • 建好圖之後在圖上跑一遍最大流即可。

代碼：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=2010;
const int MAXM=1200010;
const int INF=0x3f3f3f3f;
struct Edge
{
	int to,next,cap,flow;
}edge[MAXM];
int tol;
int head[MAXN];
void init()
{
	tol = 2;
	memset(head,-1,sizeof(head));
}
void addedge(int u,int v,int w,int rw=0)
{
	edge[tol].to=v;
	edge[tol].cap=w;
	edge[tol].flow=0;
	edge[tol].next=head[u];
	head[u]=tol++;
	edge[tol].to=u;
	edge[tol].cap=rw;
	edge[tol].flow=0;
	edge[tol].next=head[v];
	head[v]=tol++;
}
int Q[MAXN];
int dep[MAXN],cur[MAXN],sta[MAXN];
bool bfs(int s,int t,int n)
{
	int front=0,tail=0;
	memset(dep,-1,sizeof(dep[0])*(n+1));
	dep[s]=0;
	Q[tail++]=s;
	while(front<tail)
	{
		int u=Q[front++];
		for(int i=head[u];i!=-1;i=edge[i].next)
		{
			int v=edge[i].to;
			if(edge[i].cap>edge[i].flow&&dep[v]==-1)
			{
				dep[v]=dep[u]+1;
				if(v==t)
				{
					return true;
				}
				Q[tail++]=v;
			}
		}
	}
	return false;
}
int dinic(int s,int t,int n)
{
	int maxflow=0;
	while(bfs(s,t,n))
	{
		for(int i=0;i<n;i++)
		{
			cur[i]=head[i];
		}
		int u=s,tail=0;
		while(cur[s]!=-1)
		{
			if(u==t)
			{
				int tp=INF;
				for(int i=tail-1;i>=0;i--)
				{
					tp=min(tp,edge[sta[i]].cap-edge[sta[i]].flow);
				}
				maxflow+=tp;
				for(int i=tail-1;i>=0;i--)
				{
					edge[sta[i]].flow+=tp;
					edge[sta[i]^1].flow-=tp;
					if(edge[sta[i]].cap-edge[sta[i]].flow==0)
					{
						tail=i;
					}
				}
				u=edge[sta[tail]^1].to;
			}
			else if(cur[u]!=-1&&edge[cur[u]].cap>edge[cur[u]].flow&&dep[u]+1==dep[edge[cur[u]].to])
			{
				sta[tail++]=cur[u];
				u=edge[cur[u]].to;
			}
			else
			{
				while(u!=s&&cur[u]==-1)
				{
					u=edge[sta[--tail]^1].to;
				}
				cur[u]=edge[cur[u]].next;
			}
		}
	}
	return maxflow;
}
int main()
{
	init();
	int n,m,k;
	scanf("%d%d%d",&n,&m,&k);
	for(int i=2;i<=n+1;++i)
	{
		int t,mon;
		scanf("%d",&t);
		for(int j=0;j<t;++j)
		{
			scanf("%d",&mon);
			addedge(i,n+1+mon,1);
		}
	}
	int s=0,t=1,med=n+m+2;
	for(int i=1;i<=m;++i)
	{
		addedge(n+1+i,t,1);
	}
	addedge(s,med,k);
	for(int i=1;i<=n;++i)
	{
		addedge(s,i+1,1);
		addedge(med,i+1,1);
	}
	printf("%d",dinic(s,t,n+m+3));
	return 0;
}