POJ 3436 ACM Computer Factory 最大流

ACM Computer Factory

 题意：一个公司有 n 个机器， 现在一个电脑有 p 个组件，每个机器一分钟能够处理Qi个电脑，前p个数代表的是每个位置的零件的状态 0 代表的是进入到这个机器的电脑 这个位置的零件不能存在 1 代表这个位置的零件一定要有，2代表这个位置的零件可以有也可以没有。 后p个数代表的是，电脑从这个机器出去之后的状态，0代表这个位置没有零件 1代表这个位置有零件。

现在求最大组装电脑的数量，并且要输出流水线状态。

代码跑一下最大流，然后对于每一个正边的流量使用的流量就是反边的流量。

代码：

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
#define Fopen freopen("_in.txt","r",stdin); freopen("_out.txt","w",stdout);
#define LL long long
#define ULL unsigned LL
#define fi first
#define se second
#define pb push_back
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define lch(x) tr[x].son[0]
#define rch(x) tr[x].son[1]
#define max3(a,b,c) max(a,max(b,c))
#define min3(a,b,c) min(a,min(b,c))
typedef pair<int,int> pll;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const LL mod =  (int)1e9+7;

const int N = 200;
const int M = N*N;
int n, m, k;
int in[N][15], out[N][15], Q[N];

int head[N], deep[N], cur[N];
int w[M], to[M], nx[M];
int tot;
void add(int u, int v, int val){
    w[tot]  = val; to[tot] = v;
    nx[tot] = head[u]; head[u] = tot++;

    w[tot] = 0; to[tot] = u;
    nx[tot] = head[v]; head[v] = tot++;
}
int bfs(int s, int t){
    queue<int> q;
    memset(deep, 0, sizeof(deep));
    q.push(s);
    deep[s] = 1;
    while(!q.empty()){
        int u = q.front();
        q.pop();
        for(int i = head[u]; ~i; i = nx[i]){
            if(w[i] > 0 && deep[to[i]] == 0){
                deep[to[i]] = deep[u] + 1;
                q.push(to[i]);
            }
        }
    }
    return deep[t] > 0;
}
int Dfs(int u, int t, int flow){
    if(u == t) return flow;
    for(int &i = cur[u]; ~i; i = nx[i]){
        if(deep[u]+1 == deep[to[i]] && w[i] > 0){
            int di = Dfs(to[i], t, min(w[i], flow));
            if(di > 0){
                w[i] -= di, w[i^1] += di;
                return di;
            }
        }
    }
    return 0;
}

int Dinic(int s, int t){
    int ans = 0, tmp;
    while(bfs(s, t)){
        for(int i = s; i <= t; i++) cur[i] = head[i];
        while(tmp = Dfs(s, t, inf)) ans += tmp;
    }
    return ans;
}
void init(){
    memset(head, -1, sizeof(head));
    tot = 0;
}
int p;
bool check(int x, int y){
    for(int i = 1;i <= p; i++){
        if(in[y][i] == 0 && out[x][i] == 1)  return false;
        if(in[y][i] == 1 && out[x][i] == 0)  return false;
    }
    return true;
}
int au[M], av[M], aw[M];
int main(){
    while(~scanf("%d%d", &p, &n)){
        init();
        for(int i = 1; i <= n; i++){
            scanf("%d", &Q[i]);
            add(i, i+n, Q[i]);
            for(int j = 1; j <= p; j++)
                scanf("%d", &in[i][j]);
            for(int j = 1; j <= p; j++)
                scanf("%d", &out[i][j]);
        }
        int s = 0, t = n*2 + 1;
        for(int i = 1; i <= n; i++){
            int f = 0;
            for(int j = 1; j <= p; j++){
                if(in[i][j] != 2) f += in[i][j];
            }
            if(f == 0) add(s, i, inf);
            f = 0;
            for(int j = 1; j <= p; j++)
                f += out[i][j];
            if(f == p) add(i+n, t, inf);
        }
        for(int i = 1; i <= n; i++){
            for(int j = 1; j <= n; j++){
                if(i != j && check(i, j))
                    add(i+n, j, inf);
            }
        }

        printf("%d ", Dinic(s, t));

        int top = 0;
        for(int i = 1; i <= n; i++){
            for(int j = head[i]; ~j; j = nx[j]){
                if(j & 1 && w[j] != 0 && to[j] != s){
                    top +=  1;
                    au[top] = to[j] - n;
                    av[top] = i;
                    aw[top] = w[j];
                }

            }
        }
        printf("%d
", top);
        for(int i = 1; i <= top; i++)
            printf("%d %d %d
", au[i], av[i], aw[i]);
    }
    return 0;
}