题意:n个城市,m条路,现在要运输K个单位的货物,对于每条路都有个系数ai,带着x单位的货物路过这条路需要支付ai*x*x的费用,每条路有个运输货物的最大容量,问运输k单位的货物所需的最小费用
以前写的费用流都是单位流量的费用,如果这次还这么写会发现第3个样例中货物全部都往ai=1的那条边跑了,但实际上两条路各跑一个更优。观察到容量很小,只有5,所以对于每一条边,把它拆成几条边,拆出来的边容量都为1,费用分别是1ai,3ai,5ai,.....为什么这样做,其实画一下就很好理解了,它避免了基本建图的那种只能走一条边的尴尬,比如样例3,它可以选择走第一条边拆出来的费用为1a的边和第二条边拆出来的费用为1a的边
第一次见到这种操作,神奇神奇
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int maxn=1010;
const int maxm=1e5+7;
const int inf=0x3f3f3f3f;
struct Node
{
int to;
int capa;
int cost;
int next;
}edge[maxm];
int cnt;
int source,sink;
int n,m,k;
int head[maxn];
int dis[maxn];
int rec[maxn];
int pre[maxn];
bool vis[maxn];
void init()
{
cnt=0;
memset(head,-1,sizeof(head));
return;
}
void add(int u,int v,int capa,int cost)
{
edge[cnt].to=v;
edge[cnt].capa=capa;
edge[cnt].cost=cost;
edge[cnt].next=head[u];
head[u]=cnt++;
edge[cnt].to=u;
edge[cnt].capa=0;
edge[cnt].cost=-cost;
edge[cnt].next=head[v];
head[v]=cnt++;
return;
}
bool spfa()
{
memset(dis,inf,sizeof(dis));
memset(pre,-1,sizeof(pre));
memset(rec,-1,sizeof(rec));
memset(vis,false,sizeof(vis));
queue<int> que;
que.push(source);
dis[source]=0;
vis[source]=true;
while(!que.empty())
{
int node=que.front();
que.pop();
vis[node]=false;
for(int i=head[node];~i;i=edge[i].next)
{
int v=edge[i].to;
if(edge[i].capa>0&&dis[v]>dis[node]+edge[i].cost)
{
dis[v]=dis[node]+edge[i].cost;
rec[v]=i;
pre[v]=node;
if(!vis[v])
{
vis[v]=true;
que.push(v);
}
}
}
}
return dis[sink]!=inf;
}
int mcmf()
{
int maxflow=0;
int mincost=0;
while(spfa())
{
int node=sink;
int minn=inf;
while(node!=source)
{
minn=min(edge[rec[node]].capa,minn);
node=pre[node];
}
maxflow+=minn;
node=sink;
while(node!=source)
{
mincost+=minn*edge[rec[node]].cost;
edge[rec[node]].capa-=minn;
edge[rec[node]^1].capa+=minn;
node=pre[node];
}
//增广到k就结束
if(maxflow>=k) break;
}
if(maxflow>=k) return mincost;
return -1;
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
while(~scanf("%d%d%d",&n,&m,&k))
{
init();
source=1;
sink=n;
for(int i=0;i<m;i++)
{
int tmp=0;
int u,v,cost,capa;
scanf("%d%d%d%d",&u,&v,&cost,&capa);
for(int j=1;j<=capa;j++)
{
add(u,v,1,(j*j-tmp)*cost);
tmp+=j*j-tmp;
}
}
printf("%d\n",mcmf());
}
return 0;
}