poj3204 Ikki's Story I - Road Reconstruction

题意：有一个国家，有一个物资提供城市（编号为0），它负责通过城市间的路线向该国首都（编号为n-1）传送物资，但这些城市间的路线上传送的物资的量是有限的，所以传到首都的物资太少，所以国王想花钱修某些城市间的道路，使之传送的物资量增多以增加首都物资获得量。为了避免浪费钱力，他想找出那些只要重建（扩建）就能使首都物资获得量增加的道路。问这样的道理有几条。

思路：既然首都的物资获得量在未扩建前有一定限制，那么对该图求最大流后，最大流对应扩建前的首都物资获得总量最大值，也就是说图中某些边的容量会减为0，否则这最大流量必定不是最大的，而我们要找的经扩建就能使最大流增大的就是这些容量为0的边（而且对于每个这样的边（也叫满流边）一定能从S经过该边到达T）。那么题目就转化成了在求最大流后的残流网络中求这样的一个满流边集x-y（边的容量为0），使得如果在残余网络中存在路径[S,X]和[Y,T]，该边集的边的条数即为所求。找满流边集很容量，关键是判断这个满流边集中的边两端点i，j，使i能由S到达且j能达到T。那么可以对S进行一次DFS（能够联通的点标记在v1中），然后将图反向后再从T DFS一遍（能够联通的点标记在v2中）。那么对于满流集合中任意一条有向边i->j,如果v1[i]和v2[j]均标记过，则满足上述性质，为题中所求的边中的一条。

#include<iostream>
using namespace std;
const int inf = 0x7fffffff;
const int MAXN = 510;
const int MAXM = MAXN*MAXN;
int map[MAXN][MAXN];
int edge[MAXN][MAXN];
int dis[MAXN];
int queue[MAXM];
int bef[MAXN];
int num[MAXN];
int v1[MAXN],v2[MAXN]; 
void set_dis(int end)
{
	int top(0),base(0);
	memset(dis,-1,sizeof(dis));
	memset(num,0,sizeof(num));
	queue[top++] = end;
	dis[end]=0;num[0] = 1;
	while(top!=base)
	{
		int pre = queue[base++];
		for(int i=end-1;i!=0;i--)
		{
			if(dis[i] != -1 || map[i][pre] == 0)
				continue;
			dis[i] = dis[pre]+1;
			queue[top++] = i;
			num[dis[i]]++;
		}
	}
}
int cmp(int x,int y)
{
	if(x<y)
		return x;
	return y;
}
int remark(int n,int a)
{
	int tmp=inf;
    for(int i=1;i!=n+1;i++)
		if(map[a][i]!=0&&dis[i]>=0)
			tmp=cmp(tmp,dis[i]+1);
    if(tmp==inf)
		tmp=n;
    return tmp;
}
int SAP(int n,int s)
{
	int j,a = s;int flow(0);
	memset(bef,-1,sizeof(bef));
	while(dis[s]<n)
	{
		for(j=1;j!=n+1;j++)
		{
			if(map[a][j]>0&&dis[a]-1==dis[j])
			{
				break;
			}
		}
		if(j<=n)
		{
			bef[j] = a;a = j;
			if(a==n)
			{
				int i = n,max(inf);
				while(bef[i]!=-1)
				{
					max = cmp(max,map[bef[i]][i]);
					i = bef[i];
				}
				i = n;
				while(bef[i]!=-1)
				{
					map[bef[i]][i]-=max;
					map[i][bef[i]]+=max;
					i = bef[i];
				}
				flow+=max;a=s;
			}
		}
		else
		{
			int x = remark(n,a);
			num[x]++;num[dis[a]]--;
            if(num[dis[a]]==0)
				return flow;//间隙优化 gap
            dis[a]=x;
            if(a!=s)
				a=bef[a]; 
		}
	}
	return flow;	
}
void dfs(int i,int v[],int n)
{
   v[i]=1;
   for(int j=1;j<n+1;j++)
    if(!v[j]&&map[i][j])
       dfs(j,v,n); 
}
void function(int n)
{
	for(int i=1;i!=n+1;i++)
	{
		for(int j=1;j!=n+1;j++)
			cout<<map[i][j]<<" ";
		cout<<endl;
	}
	int num(0);
	dfs(1,v1,n);
	for(int i=1;i<n+1;i++)
		for(int j=i;j<n+1;j++)
			swap(map[i][j],map[j][i]); 
	for(int i=1;i!=n+1;i++)
	{
		for(int j=1;j!=n+1;j++)
			cout<<map[i][j]<<" ";
		cout<<endl;
	}
	dfs(n,v2,n);
	for(int i=1;i<n+1;i++) 
		for(int j=1;j<n+1;j++) 
			if(v1[i]&&v2[j]&&map[j][i]==0&&edge[i][j])
				num++;
	printf("%d\n",num); 
} 
int main()
{
	int e,n;
	//cout<<inf<<endl;
	while(cin>>n>>e)
	{
		memset(v1,0,sizeof(v1));
		memset(v2,0,sizeof(v2));
		memset(edge,0,sizeof(edge));
		memset(map,0,sizeof(map));
		while(e--)
		{
			int a,b,w;
			cin>>a>>b>>w;
			edge[a+1][b+1]=map[a+1][b+1] += w;
		}
		set_dis(n);
		SAP(n,1);
		function(n);
	}
	return 0;
}
/*
5 7
0 1 5
0 3 2
1 2 3
1 4 3
2 4 1
3 4 6
3 1 7
4 5
0 1 4
1 3 15
2 1 2
1 2 3
0 2 3
4 5
0 1 100
1 3 15
2 1 2
1 2 3
0 2 3
*/