Transportation HDU 3667 UVALive - 5095 拆边+最小费用最大流

传送门：HDU3667

题意：有n个点m条边组成的有向图，要把k单位的货物从1号点运送到n号点，每条路有一个系数a，重量为x的货物花费为a*x^2,问最少花费是多少。

思路：典型的最小费用最大流，难处理的地方就是边权问题，最小费用最大流一般解决的都是单位货位花费固定的问题，这样2次方增长花费的我们就不能直接套模板了。

考虑拆边，将每条边拆成容量为1，单位花费为1*a,3*a,5*a,7*a,9*a...的边，这样如果在原图中一条边上运送2单位的货物，花费为a*（2^2）=4a,在新图中我们会从拆出来的对应两条边上分别运送1单位的货物，花费为（1*a+3*a）=4a。

另外注意一点，题中已经将最大流限制为k了。

代码：

#include<bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int MAXN=110;
const int MAXM=100010;
int head[MAXN],pre[MAXN],dis[MAXN],book[MAXN];
int cnt=0,N;//N为点数 
struct node{
	int to,next,cap,flow,cost;
}edge[MAXM];
void init()
{
	cnt=0;
	memset(head,-1,sizeof(head));
}
void addedge(int u,int v,int cap,int cost)
{
	edge[cnt].to=v;edge[cnt].cap=cap;edge[cnt].flow=0;
	edge[cnt].cost=cost;edge[cnt].next=head[u];head[u]=cnt++;
	edge[cnt].to=u;edge[cnt].cap=0;edge[cnt].flow=0;
	edge[cnt].cost=-cost;edge[cnt].next=head[v];head[v]=cnt++;
}
bool spfa(int s,int t)
{
	queue<int>q;
	while(!q.empty())q.pop();
	for(int i=0;i<=N;i++)
	{
		dis[i]=inf;
		pre[i]=-1;
		book[i]=0;
	}
	dis[s]=0;book[s]=1;q.push(s);
	while(!q.empty())
	{
		int u=q.front();q.pop();book[u]=0;
		for(int i=head[u];i!=-1;i=edge[i].next)
		{
			int v=edge[i].to;
			if(edge[i].cap>edge[i].flow&&dis[v]>dis[u]+edge[i].cost)
			{
				dis[v]=dis[u]+edge[i].cost;
				pre[v]=i;
				if(!book[v])
				{
					book[v]=1;
					q.push(v);
				}
			}
		 } 
	}
	return pre[t]!=-1;
}
int min_cost_max_flow(int s,int t,int &cost,int max_flow)
{
	int flow=0;
	cost=0;
	while(spfa(s,t))
	{
		int temp=inf;
		for(int i=pre[t];i!=-1;i=pre[edge[i^1].to])
		temp=min(temp,edge[i].cap-edge[i].flow);
		for(int i=pre[t];i!=-1;i=pre[edge[i^1].to])
		{
			edge[i].flow+=temp;
			edge[i^1].flow-=temp;
			cost+=edge[i].cost*temp;
		}
		flow+=temp;
		if(flow>=max_flow)
		return flow;
	}
	return flow;
}
int main()
{
    int m,k,cost;
    int u,v,a,c;
    while(cin>>N>>m>>k)
    {
    	init();
    	for(int i=0;i<m;i++)
    	{
    		scanf("%d%d%d%d",&u,&v,&a,&c);
    		for(int j=1;j<=c;j++)//拆边
    		{
    			addedge(u,v,1,a*(j*j-(j-1)*(j-1)));
    		}
    	}
    	if(min_cost_max_flow(1,N,cost,k)>=k)
    	cout<<cost<<endl;
    	else
    	cout<<-1<<endl;
    }
    return 0;
}