【bzoj2055】80人环游世界 有源汇上下界费用流

Description

想必大家都看过成龙大哥的《80天环游世界》，里面的紧张刺激的打斗场面一定给你留下了深刻的印象。现在就有这么
一个80人的团伙，也想来一次环游世界。
他们打算兵分多路，游遍每一个国家。
因为他们主要分布在东方，所以他们只朝西方进军。设从东方到西方的每一个国家的编号依次为1...N。假若第i个人的游历路线为P1、P2......Pk(0≤k≤N)，则P1

有Vi个人会经过那一个国家。
为了节省时间，他们打算通过坐飞机来完成环游世界的任务。同时为了省钱，他们希望总的机票费最小。
明天就要出发了，可是有些人临阵脱逃，最终只剩下了M个人去环游世界。他们想知道最少的总费用，你能告诉他们吗？

Input

第一行两个正整数N，M。
第二行有N个不大于M正整数，分别表示V1，V2……VN。
接下来有N-1行。第i行有N-i个整数，该行的第j个数表示从第i个国家到第i+j个国家的机票费（如果该值等于-1则表示这两个国家间没有通航）。

Output

在第一行输出最少的总费用。

Sample Input

6 3

2 1 3 1 2 1

2 6 8 5 0

8 2 4 1

6 1 0

4 -1

4

Sample Output

27

HINT

1<= N < =100 1<= M <= 79

Source

---

拆点，i为入点，i’为出点。新建源点s，汇点e，附属源点s’。

建（s,s’,0,m,0）限制流量，（s’,i,0,INF,0）,（i’,e,0,INF,0），（i,i’,v[i],v[i],INF）化点为边,（i’,j,0,INF,cost[i,j]）。

注意边的容量是无限大的。

我决定以后费用流不写我那奇怪的dinic了……EK吧EK吧…

#include
#include
#include
#include
#include
using namespace std;

const int INF = 1000000010;
const int SZ = 100010;

int B[SZ];

int head[SZ],nxt[SZ],tot = 1;

struct edge{
    int f,t,d,c;
}l[SZ];

void build(int f,int t,int d,int c)
{
    l[++ tot].t = t;
    l[tot].f = f;
    l[tot].d = d;
    l[tot].c = c;
    nxt[tot] = head[f];
    head[f] = tot;
}

void insert(int f,int t,int d,int c)
{
    build(f,t,d,c);
    build(t,f,0,-c);
}

int dist[SZ];
queue<int> q;
bool use[SZ];

int pre[SZ];

bool spfa(int s,int e)
{
    memset(dist,63,sizeof(dist));
    dist[s] = 0;
    q.push(s);
    use[s] = 1;
    while(q.size())
    {
        int u = q.front(); q.pop();
        use[u] = 0;
        for(int i = head[u];i;i = nxt[i])
        {
            int v = l[i].t;
            if(dist[v] > dist[u] + l[i].c && l[i].d)
            {
                dist[v] = dist[u] + l[i].c;
                pre[v] = i;
                if(!use[v])
                    use[v] = 1,q.push(v);
            }
        }
    }
    if(dist[e] > INF) return false;
    return true;
}


int dfs(int s,int e)
{
    int x = INF,ans = 0;
    for(int i = pre[e];i;i = pre[l[i].f])
        x = min(x,l[i].d);
    for(int i = pre[e];i;i = pre[l[i].f])
    {
    //  printf("%d\n",l[i].t);
    //  printf("%d %d %d\n",l[i].c,B[i],x);
        ans += l[i].c * x,l[i].d -= x,l[i ^ 1].d += x;
    }
    return ans;
}

int EK(int s,int e)
{
    int ans = 0;
    while(spfa(s,e))
    {
        int tmp = dfs(s,e);
        ans += tmp;
    }
    return ans;
}

int inb[SZ],outb[SZ];



int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i = 1;i <= n;i ++)
    {
        int x;
        scanf("%d",&x);
        outb[i] += x;
        inb[i + n] += x;
        insert(i,i + n,0,0);
    }

    for(int i = 1;i <= n;i ++)
    {
        for(int j = i + 1;j <= n;j ++)
        {
            int x;
            scanf("%d",&x);
            if(x == -1) continue;
            insert(i + n,j,INF,x);
        }
    }
    int s1 = n * 2 + 1,s = n * 2 + 2,e = n * 2 + 3;

    for(int i = 1;i <= n;i ++)  insert(s1,i,INF,0);
    for(int i = 1;i <= n;i ++) insert(i + n,e,INF,0);
    insert(s,s1,m,0);   

    insert(e,s,INF,0);

    int S = n * 2 + 4,E = n * 2 + 5;
    for(int i = 1;i <= n * 2 + 3;i ++)
    {
        int tmp = inb[i] - outb[i];
        if(tmp > 0)
            insert(S,i,tmp,0);
        else
            insert(i,E,-tmp,0);
    }
    int ans = EK(S,E);

    printf("%d",ans);
    return 0;
}