P4013 数字梯形问题（网络流最大费用最大流）

题意：

给定一个由 nn 行数字组成的数字梯形如下图所示。

梯形的第一行有 mm 个数字。从梯形的顶部的 mm 个数字开始，在每个数字处可以沿左下或右下方向移动，形成一条从梯形的顶至底的路径。

分别遵守以下规则：

1. 从梯形的顶至底的 mm 条路径互不相交；

2. 从梯形的顶至底的 mm 条路径仅在数字结点处相交；

3. 从梯形的顶至底的 mm 条路径允许在数字结点相交或边相交。

思路：

建图很简单，但注意清空。

#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int N = 1e5 + 10;
int n, m, s, t, h[N], cnt, dis[N], inq[N], pre[N], last[N], flow[N];
struct node {
    int v, w, c, nt;
} no[N];
void add(int u, int v, int w, int c) {
    no[cnt] = node{v, w, c, h[u]};
    h[u] = cnt++;
    no[cnt] = node{u, 0, -c, h[v]};
    h[v] = cnt++;
}
int spfa() {
    memset(inq, 0, sizeof inq);
    memset(dis, INF, sizeof dis);
    dis[s] = 0, inq[s] = 1, flow[s] = INF;
    queue<int> q;
    q.push(s);
    while(!q.empty()) {
        int u = q.front();
        q.pop(), inq[u] = 0;
        for(int i = h[u]; ~i; i = no[i].nt) {
            int v = no[i].v;
            if(no[i].w > 0 && dis[v] > dis[u] + no[i].c) {
                dis[v] = dis[u] + no[i].c;
                pre[v] = u, last[v] = i;
                flow[v] = min(flow[u], no[i].w);
                if(!inq[v])
                    inq[v] = 1, q.push(v);
            }
        }
    }
    return dis[t] != INF;
}
ll mcmf() {
    ll ans = 0;
    while(spfa()) {
        for(int i = t; i != s; i = pre[i]) {
            no[last[i]].w -= flow[t];
            no[last[i] ^ 1].w += flow[t];
        }
        ans += 1ll * flow[t] * dis[t];
    }
    return ans;
}
int mp[100][100];
int co(int x, int y) {
    return (x - 1) * 40 + y;
}
int main() {
    memset(h, -1, sizeof h);
    scanf("%d%d", &m, &n);
    int num = m;
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= num; j++)
            scanf("%d", &mp[i][j]), add(co(i, j), co(i, j) + 1000, 1, -mp[i][j]);
        num++;
    }
    num = m;
    for(int i = 1; i <= n; i++) {
        if(i != n)
            for(int j = 1; j <= num; j++) {
                add(co(i, j) + 1000, co(i + 1, j), 1, 0);
                add(co(i, j) + 1000, co(i + 1, j + 1), 1, 0);
            }
        num++;
    }
    t = 2000;
    for(int i = 1; i <= m; i++)
        add(s, co(1, i), 1, 0);
    for(int i = 1; i < num; i++)
        add(co(n, i) + 1000, t, 1, 0);
    printf("%d\n", -mcmf());
    memset(h, -1, sizeof h);
    cnt = 0, num = m;
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= num; j++)
            add(co(i, j), co(i, j) + 1000, INF, -mp[i][j]);
        num++;
    }
    num = m;
    for(int i = 1; i <= n; i++) {
        if(i != n)
            for(int j = 1; j <= num; j++) {
                add(co(i, j) + 1000, co(i + 1, j), 1, 0);
                add(co(i, j) + 1000, co(i + 1, j + 1), 1, 0);
            }
        num++;
    }
    for(int i = 1; i <= m; i++)
        add(s, co(1, i), 1, 0);
    for(int i = 1; i < num; i++)
        add(co(n, i) + 1000, t, INF, 0);
    printf("%d\n", -mcmf());
    memset(h, -1, sizeof h);
    cnt = 0, num = m;
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= num; j++)
            add(co(i, j), co(i, j) + 1000, INF, -mp[i][j]);
        num++;
    }
    num = m;
    for(int i = 1; i <= n; i++) {
        if(i != n)
            for(int j = 1; j <= num; j++) {
                add(co(i, j) + 1000, co(i + 1, j), INF, 0);
                add(co(i, j) + 1000, co(i + 1, j + 1), INF, 0);
            }
        num++;
    }
    for(int i = 1; i <= m; i++)
        add(s, co(1, i), 1, 0);
    for(int i = 1; i < num; i++)
        add(co(n, i) + 1000, t, INF, 0);
    printf("%d\n", -mcmf());
    return 0;
}