POJ - 3436 ACM Computer Factory 最大流问题

题意

题目超长 题意超难理解 翻译一下：
输入的三项分别为, 每小时加工零件数量, 需要零件, 加工后所包含的零件
问把空零件的电脑加工为满零件的电脑 怎样搭配使得生产效率最高
即0 0 0 … 0 加工为 1 1 1 … 1

思路

最大流＋拆点问题
拆点 每个机器输入指向输出 边权为生产效率
先对于所有输入为000的建一条由源点S指向的边
再对于所有输出为111的建一条指向T的边
再遍历所有机器 如果u的输出和v的输入可以搭配 则连一条u->v的边
跑一遍dinic求最大流
找边的话可以用st来标记机器之间的边 然后再遍历所有边看st是否为true且残留网络反向边边权不为0(当前边有流)

代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>

using namespace std;

const int N = 110, M = 100010, INF = 1e8;

int n, m, S, T;
int h[N], e[M], ne[M], f[M], idx;
int cur[N], q[N], d[N];

int num[N], input[N][N], output[N][N];
bool st[N][N];

void add(int a, int b, int c)
{
    e[idx] = b, ne[idx] = h[a], f[idx] = c, h[a] = idx ++ ;
    e[idx] = a, ne[idx] = h[b], f[idx] = 0, h[b] = idx ++ ;
}

bool bfs()
{
    memset(d, -1, sizeof d);
    int hh = 0, tt = 0;
    d[S] = 0, q[0] = S, cur[S] = h[S];
    while (hh <= tt)
    {
        int t = q[hh ++ ];
        for (int i = h[t]; ~i; i = ne[i])
        {
            int ver = e[i];
            if (d[ver] == -1 && f[i])
            {
                d[ver] = d[t] + 1;
                cur[ver] = h[ver];
                if (ver == T) return true;
                q[ ++ tt] = ver;
            }
        }
    }
    return false;
}

int find(int u, int limit)
{
    if (u == T) return limit;
    int flow = 0;
    for (int i = cur[u]; ~i && flow < limit; i = ne[i])
    {
        int ver = e[i];
        cur[u] = i;
        if (d[ver] == d[u] + 1 && f[i])
        {
            int t = find(ver, min(f[i], limit - flow));
            if (!t) d[ver] = -1;
            f[i] -= t, f[i ^ 1] += t, flow += t;
        }
    }
    return flow;
}

int dinic()
{
    int r = 0, flow;
    while (bfs()) while (flow = find(S, INF)) r += flow;
    return r;
}

bool check(int u, int v)
{
    if (u == v || v == 2) return true;
    return false;
}

bool ck(int u[], int x)
{
    for (int i = 1; i <= m; i ++ )
        if (u[i] == x) return false;
    return true;
}

int main()
{
    while (~scanf("%d%d", &m, &n))
    {
        for (int i = 1; i <= n; i ++ ) 
        {
            scanf("%d", &num[i]);
            for (int j = 1; j <= m; j ++ ) scanf("%d", &input[i][j]);
            for (int j = 1; j <= m; j ++ ) scanf("%d", &output[i][j]);
        }
        
        S = 0, T = N - 1;
        
        memset(st, 0, sizeof st);
        memset(h, -1, sizeof h);
        idx = 0;
        
        // 不同机器之间的建边
        for (int i = 1; i <= n; i ++ )
            for (int j = 1; j <= n; j ++ )
            {
                if (i == j) continue;
                bool flag = true;
                for (int k = 1; k <= m; k ++ )
                    if (!check(output[i][k], input[j][k]))
                    {
                        flag = false;
                        break;
                    }
                if (flag) 
                {
                    st[i + n][j] = true;
                    add(i + n, j, num[i]);
                }
            }
        
        // 拆点以及连接源点和汇点
        for (int i = 1; i <= n; i ++ )
        {
            add(i, i + n, num[i]);
            if (ck(input[i], 1)) add(S, i, num[i]);
            if (ck(output[i], 0)) add(i + n, T, num[i]);
        }
        
        printf("%d ", dinic());
        
        int cnt = 0;
        for (int i = 0; i < idx; i += 2)
        {
            int a = e[i ^ 1], b = e[i];
            if (st[a][b] && f[i ^ 1]) cnt ++ ;
        }
        
        printf("%d\n", cnt);
        
        for (int i = 0; i < idx; i += 2)
        {
            int a = e[i ^ 1], b = e[i];
            if (st[a][b] && f[i ^ 1])
                printf("%d %d %d\n", a - n, b, f[i ^ 1]);
        }
        
    }
    
    return 0;
}