P4015 运输问题

yyc大佬对我说，网络流更应该注重对模型的学习，推荐给我网络流24题
这是第一题，加油，奥利给！！！

题意：m个仓库，n个零售商店。第i个仓库有a[i]个单位的货物，第j个零售商店需要b[j]个单位的货物。从第i个仓库到第j个零售商店的单位费用为cij。问运输方式，求最小运输费用和最大运输费用。

思路：单位费用，想最小费用最大流。
第二个点就是a[i]单位货物和零售商店只需要b[i]单位的货物。这算是一个限制条件吧，我们应该如何转化呢。而且此时我们还没有源点和汇点。我们自己建一个源点和汇点。源点连接仓库，需要流量为a[i]，费用为0；零售商店连接汇点，需要流量为b[j],费用为0，仓库连接零售商店，流量为INF，费用为cij。
最大运输费用，只需把所有费用边建成负的，结果再取相反数即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn=100010;
bool vis[maxn];
int a[1000],b[1000],ff[1000][1000];
int n,m,s,t,x,y,z,f,dis[maxn],pre[maxn],last[maxn],flow[maxn],maxflow,mincost;
//dis最小花费;pre每个点的前驱；last每个点的所连的前一条边；flow源点到此处的流量 
struct Edge{
	int to,next,flow,dis;//flow流量 dis花费 
}edge[maxn];
int head[maxn],num_edge; 
queue <int> q;

void add_edge(int from,int to,int flow,int dis)
{
	edge[++num_edge].next=head[from];
	edge[num_edge].to=to;
	edge[num_edge].flow=flow;
	edge[num_edge].dis=dis;
	head[from]=num_edge;
}

bool spfa(int s,int t)
{
	memset(dis,0x7f,sizeof(dis));
	memset(flow,0x7f,sizeof(flow));
	memset(vis,0,sizeof(vis));
	q.push(s); vis[s]=1; dis[s]=0; pre[t]=-1;
	
	while (!q.empty())
	{
		int now=q.front();
		q.pop();
		vis[now]=0;
		for (int i=head[now]; i!=-1; i=edge[i].next)
		{
			if (edge[i].flow>0 && dis[edge[i].to]>dis[now]+edge[i].dis)//正边 
			{
				dis[edge[i].to]=dis[now]+edge[i].dis;
				pre[edge[i].to]=now;
				last[edge[i].to]=i;
				flow[edge[i].to]=min(flow[now],edge[i].flow);//
				if (!vis[edge[i].to])
				{
					vis[edge[i].to]=1;
					q.push(edge[i].to);
				}
			}
		}
	}
	return pre[t]!=-1;
}

void MCMF()
{
	while (spfa(s,t))
	{
		int now=t;
		maxflow+=flow[t];
		mincost+=flow[t]*dis[t];
		while (now!=s)
		{//从源点一直回溯到汇点 
			edge[last[now]].flow-=flow[t];//flow和dis容易搞混 
			edge[last[now]^1].flow+=flow[t];
			now=pre[now];
		}
	}
}

int main()
{
	memset(head,-1,sizeof(head)); num_edge=-1;//初始化 
	scanf("%d%d",&n,&m);
	s=0;t=n+m+1;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		add_edge(s,i,a[i],0);add_edge(i,s,0,0);
	}
	for(int i=n+1;i<=n+m;i++)
	{
		scanf("%d",&b[i]);
		add_edge(i,t,b[i],0);add_edge(t,i,0,0);
	}
	for(int i=1;i<=n;i++)
	for(int j=n+1;j<=n+m;j++)
	{
		scanf("%d",&ff[i][j]);
		add_edge(i,j,0x3f3f3f3f,ff[i][j]);add_edge(j,i,0,-ff[i][j]);
	}
	MCMF();
	int pp=mincost;
	printf("%d\n",mincost);
	memset(head,-1,sizeof(head));
	for(int i=1;i<=n;i++)
	{
		add_edge(s,i,a[i],0);add_edge(i,s,0,0);
	}
	for(int i=n+1;i<=n+m;i++)
	{
		add_edge(i,t,b[i],0);add_edge(t,i,0,0);
	}
	for(int i=1;i<=n;i++)
	for(int j=n+1;j<=n+m;j++)
	{
		add_edge(i,j,0x3f3f3f3f,-ff[i][j]);add_edge(j,i,0,ff[i][j]);
	}
	MCMF();
	printf("%d\n",pp+abs(mincost));
	return 0;
}