HDU3667 Transportation —— 最小费用流（费用与流量平方成正比）

题目链接：https://vjudge.net/problem/HDU-3667

Transportation

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 3083    Accepted Submission(s): 1341

Problem Description

There are N cities, and M directed roads connecting them. Now you want to transport K units of goods from city 1 to city N. There are many robbers on the road, so you must be very careful. The more goods you carry, the more dangerous it is. To be more specific, for each road i, there is a coefficient a_i. If you want to carry x units of goods along this road, you should pay a_i * x² dollars to hire guards to protect your goods. And what’s worse, for each road i, there is an upper bound C_i, which means that you cannot transport more than C_i units of goods along this road. Please note you can only carry integral unit of goods along each road.
You should find out the minimum cost to transport all the goods safely. 

 

Input

There are several test cases. The first line of each case contains three integers, N, M and K. (1 <= N <= 100, 1 <= M <= 5000, 0 <= K <= 100). Then M lines followed, each contains four integers (u_i, v_i, a_i, C_i), indicating there is a directed road from city u_i to v_i, whose coefficient is a_i and upper bound is C_i. (1 <= u_i, v_i <= N, 0 < a_i <= 100, C_i <= 5)

 

Output

Output one line for each test case, indicating the minimum cost. If it is impossible to transport all the K units of goods, output -1.

 

Sample Input

2 1 2 1 2 1 2 2 1 2 1 2 1 1 2 2 2 1 2 1 2 1 2 2 2

 

Sample Output

4 -1 3

 

Source

2010 Asia Regional Harbin

 

Recommend

lcy

题解：

费用与流量平方成正比。详情在《训练指南》P366 。主要方法是拆边。

代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <string>
#include <set>
using namespace std;
typedef long long LL;
const int INF = 2e9;
const LL LNF = 9e18;
const int mod = 1e9+7;
const int MAXM = 1e5+10;
const int MAXN = 1e2+10;

struct Edge
{
    int to, next, cap, flow, cost;
}edge[MAXM];
int tot, head[MAXN];
int pre[MAXN], dis[MAXN];
bool vis[MAXN];
int N;

void init(int n)
{
    N = n;
    tot = 0;
    memset(head, -1, sizeof(head));
}

void add(int u, int v, int cap, int cost)
{
    edge[tot].to = v;  edge[tot].cap = cap;  edge[tot].cost = cost;
    edge[tot].flow = 0;   edge[tot].next = head[u];   head[u] = tot++;
    edge[tot].to = u;   edge[tot].cap = 0;  edge[tot].cost = -cost;
    edge[tot].flow = 0; edge[tot].next = head[v];   head[v] = tot++;
}

bool spfa(int s, int t)
{
    queue<int>q;
    for(int i = 0; i<N; i++)
    {
        dis[i] = INF;
        vis[i] = false;
        pre[i] = -1;
    }

    dis[s] = 0;
    vis[s] = true;
    q.push(s);
    while(!q.empty())
    {
        int u  = q.front();
        q.pop();
        vis[u] = false;
        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(!vis[v])
                {
                    vis[v] = true;
                    q.push(v);
                }
            }
        }
    }
    if(pre[t]==-1) return false;
    return true;
}

int minCostMaxFlow(int s, int t, int &cost)
{
    int flow = 0;
    cost = 0;
    while(spfa(s,t))
    {
        int Min = INF;
        for(int i = pre[t]; i!=-1; i = pre[edge[i^1].to])
        {
            if(Min>edge[i].cap-edge[i].flow)
                Min = edge[i].cap-edge[i].flow;
        }
        for(int i = pre[t]; i!=-1; i = pre[edge[i^1].to])
        {
            edge[i].flow += Min;
            edge[i^1].flow -= Min;
            cost += edge[i].cost*Min;
        }
        flow += Min;
    }
    return flow;
}

int main()
{
    int n, m, K;
    while(scanf("%d%d%d", &n, &m, &K)!=EOF)
    {
        init(n+1);
        for(int i = 1; i<=m; i++)
        {
            int u, v, c, a;
            scanf("%d%d%d%d", &u, &v, &a, &c);
            for(int j = 1; j<=c; j++)   //拆边
                add(u, v, 1, (j*2-1)*a);    //拆成费用为1 3 5 7 9……的边，每条边的容量为1
        }
        add(0, 1, K, 0);

        int min_cost;
        int start = 0, end = n;
        int max_flow = minCostMaxFlow(start, end, min_cost);

        if(max_flow<K) printf("-1
");
        else printf("%d
", min_cost);
    }
}