poj 3436 ACM Computer Factory

题目链接：http://poj.org/problem?id=3436

纠结了好几天的题目，看懂题意就看了好长时间，题目大意：

有N台机器，每台机器有P部分，每部分有自己的输入、输出，因此每台机器有2*P+1种参数：第一个参数Q：该机器的容量；接下来P个参数S：该机器各部分的输入；接下来P个参数D：该机器各部分的输出。

其中输入有三种情况：0,1,2

0：该部分必须不要；1：该部分必须要；2：该部分可有可无

输出有2种情况：0,1  

0：该部分不存在；1：该部分存在

题目要求的是最大流、流量改变的边数和改变的边。

思路：

如果某个节点 u 的输入部分没有1，则添加一条 s->u 路径，容量为Qu；如果某个节点 v 输出全为1，则添加一条 v->t 路径，容量为Qv；如果节点 u 的输出与 v 的输入不冲突，则添加一条 u->v 的路径，容量为min(Qu, Qv)。

看了一下网上的题解，基本上都是用拆点建图过的，不知道为什么，按照我的这个思路，没拆点也过了，可能数据弱？

代码：

const int maxn = 110;
const int maxm = 20500;
const int oo = 1<<29;
struct edge{
    int u;
    int v;
    int w;
    int next;
    edge(){
        w = 0;
    }
}edge[maxn], edge1[maxn];
int n, cnt;
int head[maxn], dep[maxn], gap[maxn];
int s[maxn], cur[maxn], head1[maxn];

void add(int u, int v, int w){
    edge[cnt].u = u;
    edge[cnt].v = v;
    edge[cnt].w = w;
    edge[cnt].next = head[u];
    edge1[cnt].u = u;
    edge1[cnt].v = v;
    edge1[cnt].w = w;
    edge1[cnt].next = head[u];
    head[u] = cnt++;
    head1[u] = cnt - 1;
    edge[cnt].u = v;
    edge[cnt].v = u;
    edge[cnt].w = 0;
    edge1[cnt].u = v;
    edge1[cnt].v = u;
    edge1[cnt].w = 0;
    edge1[cnt].next = head[v];
    edge[cnt].next = head[v];
    head[v] = cnt++;
    head1[v] = cnt - 1;
}

void BFS(int start, int end){
    int i, to, u;
    memset(dep, -1, sizeof(dep));
    memset(gap, 0, sizeof(gap));
    queue <int> que;
    dep[end] = 0;
    gap[0] = 1;
    que.push(end);
    while(!que.empty()){
        u = que.front();
        que.pop();
        for(i = head[u]; i != -1; i = edge[i].next){
            to = edge[i].v;
            if(dep[to] != -1 || edge[i].w != 0) continue;
            que.push(to);
            dep[to] = dep[u]+1;
            gap[dep[to]]++;
        }
    }
}

int sap(int start, int end){
    int res = 0;
    BFS(start, end);
    int u = start, top = 0;
    memcpy(cur, head, sizeof(head));
    while(dep[start] < n){
        if(u == end){
            int Min = oo, flag;
            for(int i = 0; i < top; i++){
                if(edge[s[i]].w < Min){
                    Min = edge[s[i]].w;
                    flag = i;
                }
            }
            res += Min;
            for(int i = 0; i < top; i++){
                edge[s[i]].w -= Min;
                edge[s[i]^1].w += Min;
            }
            top = flag;
            u = edge[s[top]].u;
        }
        if(u != end && gap[dep[u]-1] == 0) break;
        int i;
        for(i = cur[u]; i != -1; i = edge[i].next){
            if(edge[i].w != 0 && dep[edge[i].v]+1 == dep[u]) break;
        }
        if(i != -1){
            cur[u] = i;
            s[top++] = i;
            u = edge[i].v;
        }
        else{
            int Min = n;
            for(i = head[u]; i != -1; i = edge[i].next){
                if(edge[i].w == 0) continue;
                if(Min > dep[edge[i].v]){
                    cur[u] = i;
                    Min = dep[edge[i].v];
                }
            }
            gap[dep[u]]--;
            dep[u] = Min+1;
            gap[dep[u]]++;
            if(u != start)
                u = edge[s[--top]].u;
        }
    }
    return res;
}

void init(){
    cnt = 0;
    memset(head, -1, sizeof(head));
}

int main(){
    #ifndef ONLINE_JUDGE
        freopen("in.txt", "r", stdin);
        //freopen("out.txt", "w", stdout);
    #endif
    int p;
    int a[maxn], s[maxn][maxn], q[maxn][maxn];
    while(~scanf("%d %d", &p, &n)){
        init();
        int pr[101][3];
        int start , end;
        start = 0;
        end = n+1;
        for(int i = 1; i <= n; i++){
            scan_d(a[i]);
            for(int j = 0; j < p; j++){
                scan_d(s[i][j]);
            }
            for(int j = 0; j < p; j++){
                scan_d(q[i][j]);
            }
        }
        for(int i = 1; i <= n; i++){
            bool fs = true, ft = true;
            for(int j = 0; j < p; j++){
                if(s[i][j] == 1) fs = false;
                if(q[i][j] == 0) ft = false;
            }
            if(fs) {
                //printf("%d %d %d\n", start, i, a[i]);
                add(start, i, a[i]);
            }
            if(ft){
                //printf("%d %d %d\n", i, end, a[i]);
                add(i, end, a[i]);
            }
            bool fc = true;
            for(int j = 1; j <= n; j++){
                if(i != j){
                    fc = true;
                    for(int k = 0; k < p && fc; k++){
                        if(q[i][k] + s[j][k] == 1) fc = false; 
                    }
                    if(fc){
                        //printf("%d %d %d\n", i, j, min(a[i], a[j]));
                        add(i, j, min(a[i], a[j]));
                    }
                }
            }
        }
        n += 2;
        int res = sap(start, end);
        //cout << res << endl;
        int numb = 0;
        for(int i = 0; i <= cnt; i++){
            if(edge[i].w < edge1[i].w && edge[i].u!=0 && edge[i].v!=n-1){
                pr[numb][0] = edge[i].u;
                pr[numb][1] = edge[i].v;
                pr[numb++][2] =  edge1[i].w - edge[i].w;
            }
        }
        printf("%d %d\n", res, numb);
        for(int i = 0; i < numb; i++){
            printf("%d %d %d\n", pr[i][0], pr[i][1], pr[i][2]);
        }
    }
    return 0;
}