POJ2516 Minimum Cost 最小费用最大流

Problem Address：http://poj.org/problem?id=2516

【思路】

对每种商品进行一个最小费用最大流的运算。

m个供应商与n个店主之间建立m*n个连接，容量无穷大，花费题目中有给出。

源点到m个供应商之间建立m个连接，花费为零，容量为供应商供应该商品的数目。

n个店主与汇点之间建立n个连接，花费为零，容量为店主需要的商品数目。

进行最小费用最大流运算，判断结果是否满足条件。满足则继续计算下一个商品，否则退出输出-1。

判断依据是每个店主最后的分配量是否能满足其需求量。

由于一半是昨晚写的，所以今天早上继续写的时候就出了点小错。

本来要用memset给d[ ]赋值MAX，后来发现不行。

输出-1忘了“\n"，拿到了一个PE。

虽然代码比较长，但是只要思路清晰，就可以慢慢写好的。

【代码】

#include <iostream>
using namespace std;

const int maxn = 50;
const int size = 100;
const int MAX = 99999999;

int shopkeeper[maxn+5][maxn+5];
int supply[maxn+5][maxn+5];
int cost[maxn+5][maxn+5][maxn+5];

int map[size+5][size+5];
int flow[size+5][size+5];
int fee[size+5][size+5];
int pre[size+5];
int d[size+5];
bool visited[size+5];
int q[size*size];


void init(int n, int m)
{
	int i,j;
	for (i=0; i<n+m+2; i++)
	{
		for (j=0; j<n+m+2; j++)
		{
			map[i][j] = 0;
		}
	}
	for (i=1; i<=n; i++)
	{
		for (j=1; j<=m; j++)
		{
			map[j+n][i] = MAX;
		}
	}
}

void cur_init(int n, int m, int k)
{
	int i,j;
	for (j=1; j<=m; j++)
	{
		map[0][j+n] = supply[j][k];
	}
	for (i=1; i<=n; i++)
	{
		map[i][n+m+1] = shopkeeper[i][k];
	}
	memset(fee, 0, sizeof(fee));
	for (j=1; j<=m; j++)
	{
		for (i=1; i<=n; i++)
		{
			fee[j+n][i] = cost[k][j][i];
			fee[i][j+n] = -fee[j+n][i];
		}
	}
	memset(flow, 0, sizeof(flow));
}

bool spfa(int s, int t, int mapsize)
{
	int i, u, v;
	int head = 1, tail = 0;
	for (i=0; i<mapsize; i++)
	{
		visited[i] = false;
		d[i] = MAX;
		pre[i] = -1;
	}
	q[0] = s;
	visited[s] = true;
	d[s] = 0;
	while(tail<head)
	{
		u = q[tail];
		tail++;
		visited[u] = false;
		for (v=0; v<mapsize; v++)
		{
			if (map[u][v]>flow[u][v] && d[u]+fee[u][v]<d[v])
			{
				d[v] = d[u] + fee[u][v];
				pre[v] = u;
				if (!visited[v])
				{
					visited[v] = true;
					q[head] = v;
					head++;
				}
			}
		}
	}
	if (pre[t]==-1) return false;
	else return true;
}

int MinCost(int s, int t, int n, int m, int k)
{
	cur_init(n, m, k);
	int res = 0;
	int u, v, inc;
	int i;
	while(spfa(s, t, n+m+2))
	{
		inc = MAX;
		for (v=t; v!=s; v=u)
		{
			u = pre[v];
			if (map[u][v]-flow[u][v]<inc)
				inc = map[u][v] - flow[u][v];
		}
		for (v=t; v!=s; v=u)
		{
			u = pre[v];
			flow[u][v] += inc;
			flow[v][u] -= inc;
		}
		res += d[t]*inc;
	}
	for (i=1; i<=n; i++)
	{
		if (map[i][t]!=flow[i][t])
			return -1;
	}
	return res;
}

int main()
{
	int n, m, kind;
	int i, j, k;
	int temp, sum;
	while(scanf("%d %d %d", &n, &m, &kind)!=EOF)
	{
		if (n==0 && m==0 && kind==0) break;
		for (i=1; i<=n; i++)
		{
			for (k=1; k<=kind; k++)
				scanf("%d", &shopkeeper[i][k]);
		}
		for (i=1; i<=m; i++)
		{
			for (k=1; k<=kind; k++)
				scanf("%d", &supply[i][k]);
		}
		memset(cost, 0, sizeof(cost));
		for (k=1; k<=kind; k++)
		{
			for (i=1; i<=n; i++)
			{
				for (j=1; j<=m; j++)
				{
					scanf("%d", &cost[k][j][i]);
				}
			}
		}
		init(n, m);
		sum = 0;
		for (k=1; k<=kind; k++)
		{
			temp = MinCost(0, n+m+1, n, m, k);
			if (temp==-1)
				break;
			sum += temp;
		}
		if (k<=kind)
			printf("-1\n");
		else
			printf("%d\n", sum);
	}
	return 0;
}