poj - 3436 ACM Computer Factory(最大流+路径打印)

第一行给你P和N, P表示配件的个数, N表示机器的数量,机器可用来组装电脑,把进来时拥有的配件变成出去时的配件。

其中进入的电脑配件有三种情况:1:已配置 0:没配置 2:配置没配置都行。

从机器中出去的电脑配件有两种情况:1:已配置 0:没配置

真的恶心哦,对于我这种第一题不做出来心里难受但是它就在第一题并且这是我第二次做最大流这题那么难这么放题合适吗?不过虽然这么说,我学到了不少东西,最起码又把链式前向星回顾了一遍。直接进入正题,也不说我犯了哪些弱智到家的错误了。

首先应该建图,怎么建图呢?我们可以把题中给出的2 * N种零件加上全没组装(源点)和全部组装完毕(汇点)当做结点,然后把零件相同的建立边,边权为INF,把生产前到生产后的情况建边,边权为C(这种机器的数量)。

然后就是求最大流。

最后输出路径,怎么输出呢?查找从这个机器出来再进入别的机器的,判断权值。如果如果权值仍为INF,就说明没有变,不然的话就输出INF - 容量,即为占用的流量,占用这个机器的流量数,也就是使用这个机器的数量。

代码如下:

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <stack>
#include <queue>
#define INF 0x3f3f3f3f
#define ll long long
 
using namespace std;
 
const int inf = 0x3f3f3f3f;
const int maxn = 1005;
const int maxm = 1000005;
 
int n, m;//点数、边数
int sp, tp;//原点、汇点
 
struct node {
    int u;
    int v, next;
    int cap;
}mp[maxm];
 
int pre[maxn], dis[maxn], cur[maxn];//cur为当前弧优化,dis存储分层图中每个点的层数(即到原点的最短距离),pre建邻接表
int cnt = 0;
 
void init() {  //不要忘记初始化
    cnt = 0;
    memset(pre, -1, sizeof(pre));
}
 
void add(int u, int v, int w) { //加边
    mp[cnt].u = u;
    mp[cnt].v = v;
    mp[cnt].cap = w;
    mp[cnt].next = pre[u];
    pre[u] = cnt++;
 
    mp[cnt].u = v;
    mp[cnt].v = u;
    mp[cnt].cap = 0;
    mp[cnt].next = pre[v];
    pre[v] = cnt++;
}
 
bool bfs() {  //建分层图
    memset(dis, -1, sizeof(dis));
    queue<int>q;
    while(!q.empty())
        q.pop();
    q.push(sp);
    dis[sp] = 0;
    int u, v;
    while(!q.empty()) {
        u = q.front();
        q.pop();
        for(int i = pre[u]; i != -1; i = mp[i].next) {
            v = mp[i].v;
            if(dis[v] == -1 && mp[i].cap > 0) {
                dis[v] = dis[u] + 1;
                q.push(v);
                if(v == tp)
                    break;
            }
        }
    }
    return dis[tp] != -1;
}
 
int dfs(int u, int cap) {//寻找增广路
    if(u == tp || cap == 0)
    return cap;
    int res = 0, f;
    for(int &i = cur[u]; i != -1; i = mp[i].next) {//
        int v = mp[i].v;
        if(dis[v] == dis[u] + 1 && (f = dfs(v, min(cap - res, mp[i].cap))) > 0) {
            mp[i].cap -= f;
            mp[i ^ 1].cap += f;
            res += f;
            if(res == cap)
                return cap;
        }
    }
    if(!res)
        dis[u] = -1;
    return res;
}
 
int dinic() {
    int ans = 0;
    while(bfs()) {
        for(int i = sp; i <= tp; i++)
            cur[i] = pre[i];
        ans += dfs(sp, inf);
    }
    return ans;
}
 
int infw[55][15];
int ofw[55][15];
 
int main()
{
    while(~scanf("%d %d", &n, &m)) {
        int c;
        init();
        sp = 0, tp = m * 2 + 1;
        for(int i = 1; i <= m; i++) {
            scanf("%d", &c);
            add(i, i + m, c);
            int flag = 1;
            for(int j = 1; j <= n; j++) {
                scanf("%d", &infw[i][j]);
                if(infw[i][j] == 1) {
                    flag = 0;
                }
            }
            if(flag) {
                add(sp, i, inf);
            }
            flag = 1;
            for(int j = 1; j <= n; j++) {
                scanf("%d", &ofw[i][j]);
                if(ofw[i][j] == 0) {
                    flag = 0;
                }
            }
            if(flag) {
                add(i + m, tp, inf);
            }
            for(int j = 1; j < i; j++) {
                flag = 1;
                for(int k = 1; k <= n; k++) {
                    if(infw[j][k] == 1 && ofw[i][k] == 0 || infw[j][k] == 0 && ofw[i][k] == 1) {
                        flag = 0;
                        break;
                    }
                }
                if(flag) {
                    add(i + m, j, inf);
                }
                flag = 1;
                for(int k = 1; k <= n; k++) {
                    if(infw[i][k] == 1 && ofw[j][k] == 0 || infw[i][k] == 0 && ofw[j][k] == 1) {
                        flag = 0;
                        break;
                    }
                }
                if(flag) {
                    add(j + m, i, inf);
                }
            }
        }
        int ans = dinic();
        int sum = 0;
        for(int i = 1; i <= m; i++) {
            for(int p = pre[i + m]; p; p = mp[p].next) {
                if(mp[p].v != i && mp[p].v != sp && mp[p].v != tp && mp[p].cap != inf) {
                    sum++;
                }
            }
        }
        printf("%d %d\n", ans, sum);
        for(int i = 1; i <= m; i++) {
            for(int p = pre[i + m]; p; p = mp[p].next) {
                if(mp[p].v != i && mp[p].v != sp && mp[p].v != tp && mp[p].cap != inf) {
                    printf("%d %d %d\n", i, mp[p].v, inf - mp[p].cap);
                }
            }
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_38367681/article/details/81390235