poj 2135 Farm Tour

昨天在做去年哈尔滨的区域赛看到了一道费用流题，才想到自己还没有做过。

于是，利用光神模版，A了第一道最小费用最大流。

很裸的，注意要加双向边，边要开8倍大小。

#include <iostream>
#define N 1005
#define M 10050*8
#define inff 0x3f3f3f3f
#define infc 0x3f3f3f3f
using namespace std;

struct mincostmaxflow
{
	int e,ev[M],nxt[M],head[N];
	int cost[M],dist[N];
	int cap[M];
	int pnt[N],road[N],q[N],bg,ed;
	bool vis[N];
	void init()
	{
		e=0;
		memset(head,-1,sizeof(head));
	}
	void addedge(int u,int v,int f,int c)
	{
		ev[e]=v;
		cap[e]=f;
		cost[e]=c;
		nxt[e]=head[u];
		head[u]=e++;
		ev[e]=u;
		cap[e]=0;
		cost[e]=-c;
		nxt[e]=head[v];
		head[v]=e++;
	}
	bool spfa(int s,int t,int n)
	{
		for(int i=0;i<=n;i++)
		{
			dist[i]=infc;
			vis[i]=0;
		}
		bg=ed=dist[s]=0;
		pnt[s]=s;
		q[ed++]=s; 
		while(bg!=ed)
		{
			int u=q[bg++];
			vis[u]=0;
			if(bg==N)
				bg=0;
			for(int i=head[u];~i;i=nxt[i])
			{
				if(cap[i]<=0)
					continue;
				int v=ev[i];
				if(dist[v]>dist[u]+cost[i])
				{
					dist[v]=dist[u]+cost[i];
					pnt[v]=u;
					road[v]=i;
					if(!vis[v])
					{
						q[ed++]=v;
						vis[v]=1;
						if(ed==N)
							ed=0;
					}
				}
			}
		}
		return dist[t]!=infc;
	}
	void mincost(int s,int t,int n,int &f,int &c)
	{
		c=f=0;
		while(spfa(s,t,n))
		{
			int minf=inff;
			for(int u=t;u!=s;u=pnt[u])
				minf=min(minf,cap[road[u]]);
			for(int u=t;u!=s;u=pnt[u])
			{
				cap[road[u]]-=minf;
				cap[road[u]^1]+=minf;
			}
			f+=minf;
			c+=minf*dist[t];
		}
	}
};
int n,m,k;
int main()
{
	int i,j,u,v,w,a,c,k,f;
	mincostmaxflow pp;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		pp.init();
		for(i=0;i<m;i++)
		{
			scanf("%d%d%d",&u,&v,&c);
			pp.addedge(u,v,1,c);
			pp.addedge(v,u,1,c);
		}
		pp.addedge(0,1,2,0);
		pp.addedge(n,n+1,2,0);
		pp.mincost(0,n+1,n+2,f,c);
		printf("%d\n",c);
	}
	return 0;
}