2237. 猪 (网络流实战)

题目
题意:
1.顾客到达养猪场，将所有他有钥匙的猪舍的大门全部打开，米尔克从所有未上锁的猪舍中挑选一定数量的猪卖给该顾客。
2.如果米尔克愿意，他还可以给未上锁的猪舍里剩下的猪重新分配位置。
3.在每个顾客到达之前，会将上一个顾客打开的猪舍全部关闭。

每个猪舍中都可以放置无限数量的猪。
请你编写一个程序，计算他当天可以出售的生猪的最大数量。
思路: 如果没有猪舍重新分配猪的限制，很好建模，直接源点向猪舍连边，猪舍向顾客连边，顾客向汇点连边，求最大流即可。但是现在可以重新分配，就比较难。可以按照天数来建点，也可以换个思路，为了显示能重新分配猪，可以由顾客向能开锁的猪舍都连一条INF的边，就可以重新分配。但是这样时序就丢了。可以进一步考虑，把猪舍删掉，如果某个猪舍是第一次被访问，源点直接向顾客连边；如果猪舍不是第一次被访问，可以由上次访问该猪舍的顾客A向该顾客B连边。因为某顾客能开锁的所有猪舍都能归他分配，所以他向别的顾客连一条INF的边，代表A能任意分配自己能开锁的猪舍的猪，把剩余的留给顾客B.(其实这样相当于所有访问过猪舍i的顾客都向当前顾客B连一条INF的边，但是根据传递性，只需要最后一次访问的顾客连边即可)

PS: 记得开明白数组，不然无缘无故地wa
代码:

#include<bits/stdc++.h>
using namespace std;
const int N = 110;
const int M = 2*(N+N*1000);
const int INF = 1e8;
typedef long long ll;
int cur[N],d[N],q[N];
int h[N],e[M],ne[M],w[M],idx = 0;
int n,m,k,T; int st,ed;
int lst[1102];
void add(int a,int b,int c)
{
	e[idx] = b,w[idx] = c,ne[idx] = h[a],h[a] = idx++;
	e[idx] = a,w[idx] = 0,ne[idx] = h[b],h[b] = idx++;
}
bool bfs()
{
	int hh = 0,tt = 0;
	memset(d,-1,sizeof(d));
	d[st] = 0,cur[st] = h[st],q[tt++] = st;
	while(hh!=tt)
	{
		int u = q[hh++];
		for(int i=h[u];~i;i=ne[i])
		{
			int j = e[i];
			if(d[j]==-1&&w[i])
			{
				d[j] = d[u]+1;
				cur[j] = h[j];
				if(j==ed) return 1;
				q[tt++] = j;
			}
		}
	}
	return 0;
}
ll find(int u,ll limit)
{
	if(u==ed) return limit;
	ll flow = 0;
	for(int i=cur[u];~i&&flow<limit;i=ne[i])
	{
		int j = e[i];
		cur[u] = i; //当前弧优化
		if(d[j]==d[u]+1&&w[i])
		{
			ll t = find(j,min(1ll*w[i],limit-flow));
			if(!t) d[j] = -1;
			flow += t,w[i] -= t,w[i^1] += t;
		}
	}
	return flow;
}
ll dinic()
{
	ll ans = 0,flow;
	while(bfs()) while(flow = find(st,INF)) ans += flow;
	return ans;
}
int a[1002];
void solve()
{
    memset(h,-1,sizeof(h));
	cin>>m>>n;
	st = 0,ed = n+1;
	for(int i=1;i<=m;++i) cin>>a[i];
	for(int i=1;i<=n;++i)
	{
		int x; cin>>x;
		while(x--)
		{
			int t; cin>>t;
			if(!lst[t]) add(st,i,a[t]); //第一次打开
			else add(lst[t],i,INF); //传递，和之前所有连过t这个猪圈的人和这个人连是等价的.
			lst[t] = i; 
		}
		int y; cin>>y;
		add(i,ed,y); //购买
	}
	ll ans = dinic();
	cout<<ans;
}
int main(void)
{
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	T = 1;
	while(T--)
	solve();
	return 0;
}
/*
2
5 3.5
1 1 1 1
2 3 0 1
3 5 1 1
5 1 1 1
5 4 0 1
3 1.1
-1 0 5 10
0 0 3 9
2 0 1 1
*/