POJ-3436 ACM Computer Factory 网络流

　　该题题义是这样的，有N台机器安排去组装电脑，每台电脑有P个零部件，一台可以出产的电脑必须要求每个部件都有。每台机器能够接受的本成品电脑不同，对于每个部件，输入数据给出了0（一定不能够在半成品上出现），1（一定要在上面出现），2（可以出现也可以不出现）。每台机器生产出来的半成品（或者成为了成品）的情况也有不同，对于每个部件0代表经过这个机器加工后，该号不见没有，相反，1代表有。

　　由于每台机器的处理能力有限，所以这里要将点拆成边，点内的边赋值为产能，其他边均赋值为无穷大。对部件没有要求的机器为源点，对部件都有要求的机器为汇点。考虑好机器与机器之间的传递关系，建图后直接计算最大流，最后在遍历两两点之间的flow。

　　代码如下：

#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <algorithm>
#include <queue>
#define SP 0
#define T 2*N+1
#define INF 0x3fffffff
#define CP(x) (N+(x))
#define MAXN 110
using namespace std;

int P, N, cap[MAXN][MAXN], flow[MAXN][MAXN];
int c[MAXN], path[MAXN], MaxFlow;

struct Node
{
    int Q, S[15], D[15];
}e[55];

void init()
{
    MaxFlow = 0;
    memset(cap, 0, sizeof (cap));
    memset(flow, 0, sizeof (flow));
}

bool Start(int x)
{
    for (int i = 1; i <= P; ++i) {
        if (e[x].S[i] == 1) {
            return false;
        }
    }
    return true;
}

bool End(int x)
{
    for (int i = 1; i <= P; ++i) {
        if (e[x].D[i] == 0) {
            return false;
        }
    }
    return true;
}

bool OK(int x, int y)
{
    for (int i = 1; i <= P; ++i) {
        if (e[x].D[i] == 1 && (e[y].S[i] == 1 || e[y].S[i] == 2)) {
            continue;
        }
        else if (e[x].D[i] == 0 && (e[y].S[i] == 0 || e[y].S[i] == 2)) {
            continue;
        }
        else {
            return false;
        }
    }
    return true;
}
/*
3 4
15  0 0 0  0 1 0
10  0 0 0  0 1 1
30  0 1 2  1 1 1
3   0 2 1  1 1 1
*/
void bfs()
{
    int pos;
    bool finish = false;
    while (!finish) {
        memset(c, 0, sizeof (c));
        queue<int>q;
        c[SP] = INF;
        q.push(SP);
        while (!q.empty()) {
            if (c[T]) {
                break;
            }
            pos = q.front();
            q.pop();
            for (int i = 1; i <= T; ++i) {
                if (!c[i] && cap[pos][i] > flow[pos][i]) { 
                    c[i] = min(c[pos], cap[pos][i]-flow[pos][i]);
                    q.push(i);
                    path[i] = pos;
                }
            }   
        }
        if (c[T] == 0) {
            finish = true;
        }
        else {
            MaxFlow += c[T];
            pos = T;
            while (pos != SP) {
                flow[path[pos]][pos] += c[T];
                flow[pos][path[pos]] -= c[T];
                pos = path[pos];
            }
        }
    }
}

void print()
{
    int M = 0;
    for (int i = 1; i <= N; ++i) {
        for (int j = 1; j <= N; ++j) {
            if (flow[CP(i)][j] > 0) {
                ++M;
            }
        }
    }
    printf("%d %d\n", MaxFlow, M);
    for (int i = 1; i <= N; ++i) {
        for (int j = 1; j <= N; ++j) {
            if (flow[CP(i)][j] > 0) {
                printf("%d %d %d\n", i, j, flow[CP(i)][j]);
            }
        }
    } 
}

int main()
{
    while (scanf("%d %d", &P, &N) == 2) {
        init();
        for (int i = 1; i <= N; ++i) {
            scanf("%d", &e[i].Q);
            for (int j = 1; j <= P; ++j) {
                // 加工的条件
                scanf("%d", &e[i].S[j]);
            }
            for (int j = 1; j <= P; ++j) {
                // 出产的特性
                scanf("%d", &e[i].D[j]);
            }
        }
        for (int i = 1; i <= N; ++i) {
            if (Start(i)) {
                cap[SP][i] = INF;
            }
            if (End(i)) { 
                cap[CP(i)][T] = INF;
            }
            cap[i][CP(i)] = e[i].Q;
            for (int j = 1; j <= N; ++j) {
                if (OK(i, j)) {
                    cap[CP(i)][j] = INF;
                }
            }
        }
        bfs();
        print();
    }
    return 0;
}