poj 2135 (基础费用流)

题意:从1到n再到1,每条边只能走一次,求最短距离。

建图:每条边只能走一次就是流量是1,添加源点与1相连,容量为2,费用为0,n与汇点相连容量为2,费用为0;

求增广路用SPFA最短路求,,






 

#include<stdio.h>
#include<queue>
#include<string.h>
const int N=1100;
const int inf=0x3fffffff;
using namespace std;
int cost[N],start,end,n,head[N],num,pre[N],vis[N];
struct edge
{
	int st,ed,cp,flow,next;
}e[N*N];
void addedge(int x,int y,int c,int w)
{
	e[num].st=x;e[num].ed=y;e[num].cp=c; e[num].flow=w;e[num].next=head[x];head[x]=num++;
	e[num].st=y;e[num].ed=x;e[num].cp=-c;e[num].flow=0;e[num].next=head[y];head[y]=num++;
}
int SPFA()
{
	int i,u,v;
	queue<int>Q;
	for(i=start;i<=end;i++)
	{cost[i]=inf;vis[i]=0;pre[i]=-1;}
	cost[start]=0;vis[start]=1;
	Q.push(start);	
	while(!Q.empty())
	{
		u=Q.front();
		Q.pop();vis[u]=0;
		for(i=head[u];i!=-1;i=e[i].next)
		{
			v=e[i].ed;
			if(e[i].flow>0&&cost[v]>cost[u]+e[i].cp)
			{
				pre[v]=i;
				cost[v]=cost[u]+e[i].cp;
				if(vis[v]==0)
				{
					vis[v]=1;
					Q.push(v);
				}
			}
		}
	}
	if(pre[end]==-1)
		return 0;
	return 1;
}
int mincost()
{
	int MINcost=0,maxflow=0,i,minflow;
	while(SPFA())
	{
		minflow=inf;
		for(i=pre[end];i!=-1;i=pre[e[i].st])
			if(minflow>e[i].flow)
				minflow=e[i].flow;
			maxflow+=minflow;
		for(i=pre[end];i!=-1;i=pre[e[i].st])
		{
			e[i].flow-=minflow;
			e[i^1].flow+=minflow;
			MINcost+=e[i].cp;
		}			
	}
	return MINcost;
}
int main()
{
	int i,x,y,c,m;
	while(scanf("%d%d",&n,&m)!=-1)
	{
		start=0;end=n+1;num=0;
		memset(head,-1,sizeof(head));
		for(i=0;i<m;i++)
		{
			scanf("%d%d%d",&x,&y,&c);
			addedge(x,y,c,1);
			addedge(y,x,c,1);
		}
		addedge(start,1,0,2);
		addedge(n,end,0,2);
		printf("%d\n",mincost());
	}
	return 0;
}


 

 

你可能感兴趣的:(poj)