POJ 3436 ACM Computer Factory (最大流)

题意：p个零件，n台机器，每台机器能生产num个零件组合，并给出每台机器的生产条件：p个数，分别代表每个零件的情况，0：不需要，1：需要，2：随便；以及生产结果：p个数，0：无，1：有。求最多能生产多少台电脑（包含所有零件），以及机器之间的协作情况。

题解：最大流
机器之间的协作情况，就是从上一台机器的生产结果，能否成为下一台机器的生产条件，即相等。我们可以遍历，然后建图，跑最大流。

源点就是000（0），遍历，找生产条件是源点的，建边。
汇点就是111（n+1），遍历，找生产结果是汇点的，建边。

这里用的sap。

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
using namespace std;
int p, n, ans[255][3];
struct node {
	int in[11], out[11];
	int num;
}mac[55];
/*
 * SAP 邻接矩阵形式
 * 点的编号从 0 开始
 * 增加个 flow 数组，保留原矩阵 maze, 可用于多次使用最大流
 */
const int MAXN = 55;
int maze[MAXN][MAXN];
int gap[MAXN], dis[MAXN], pre[MAXN], cur[MAXN];
int flow[MAXN][MAXN];//存最大流的流量
int sap(int start, int end, int nodenum) {
    memset(cur, 0, sizeof(cur));
    memset(dis, 0, sizeof(dis));
    memset(gap, 0, sizeof(gap));
    memset(flow, 0, sizeof(flow));
    int u = pre[start] = start, maxflow = 0, aug = -1;
    gap[0] = nodenum;
    while (dis[start] < nodenum) {
    loop:
        for (int v = cur[u]; v < nodenum; v++) {
            if (maze[u][v] - flow[u][v] && dis[u] == dis[v] + 1) {
                if (aug == -1 || aug > maze[u][v] - flow[u][v]) aug = maze[u][v] - flow[u][v];
                pre[v] = u;
                u = cur[u] = v;
                if (v == end) {
                    maxflow += aug;
                    for (u = pre[u]; v != start; v = u, u = pre[u]) {
                        flow[u][v] += aug;
                        flow[v][u] -= aug;
                    }
                    aug = -1;
                }
                goto loop;
            }
        }
        int mindis = nodenum - 1;
        for (int v = 0; v < nodenum; v++) {
            if (maze[u][v] - flow[u][v] && mindis > dis[v]) {
                cur[u] = v;
                mindis = dis[v];
            }
        }
        if ((--gap[dis[u]]) == 0) break;
        gap[dis[u] = mindis + 1]++;
        u = pre[u];
    }
    return maxflow;
}
int main() {
	scanf("%d%d", &p, &n);
	for (int i = 1; i <= n; i++) {
		scanf("%d", &mac[i].num);                                 //生产数量
		for (int j = 1; j <= p; j++) scanf("%d", &mac[i].in[j]);  //生产条件
		for (int j = 1; j <= p; j++) scanf("%d", &mac[i].out[j]); //生产结果
	}
    for (int i = 1; i <= n; i++) {
        int f1 = 1, f2 = 1;
        for (int j = 1; j <= p; j++) {
            if (mac[i].in[j] == 1) f1 = 0;
            if (mac[i].out[j] == 0) f2 = 0;
        }
        if (f1) maze[0][i] = mac[i].num;       //源点 起始都是000
        if (f2) maze[i][n + 1] = mac[i].num;   //汇点 终止都是111
        for (int j = 1; j <= n; j++) {         //将该机器与其他机器匹配
            if (i == j) continue;
            int f = 1;
            for (int k = 1; k <= p; k++) {
                if (mac[i].out[k] + mac[j].in[k] == 1) {  //结果等于下一个机器的条件
                    f = 0;
                    break;
                }
            }
            if (f) maze[i][j] = min(mac[i].num, mac[j].num);
        }
    }
    int maxflow = sap(0, n + 1, n + 2);
    int cnt = 0;
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            if (flow[i][j] > 0) {
                ans[++cnt][0] = i;
                ans[cnt][1] = j;
                ans[cnt][2] = flow[i][j];
            }
        }
    }
    printf("%d %d\n", maxflow, cnt);
    for (int i = 1; i <= cnt; i++) printf("%d %d %d\n", ans[i][0], ans[i][1], ans[i][2]);
	return 0;
}