【最大流 && 输出路径】POJ - 3436 ACM Computer Factory

Problem Description

输入p, n代表一个机器由p个部件(0代表没有该部件，1代表有)，n代表有n种转换方式。接下来n行每行输入，a[i], in[i][1] … in[i][p], out[i][1]…out[i][p]。例如：15 0 0 0 0 1 0，机器人身上的部件满足0 0 0可以转换成0 1 0，最大可以转换15台。问你最后产生1 1 1的机器人有几台。同时输出路径

思路：

关键点就在于建图，还有输出路径。建图：我们可以把0 0 0看成一个点，0 1 0看成一个点。如果0 0 0 是i点，那么0 1 0就是i + n这个点。那么i这个点到i + n这个点的流量就是a[i]。所以1 + n..n+n的点到1…n的点，如果机器人部件都一样，那么流量就是无穷。内部图已经建好了。接下来添加一个超级源点0，和超级汇点2*n+1。超级源点到i…n的点的流量为无穷。i+n…n + n的点中（机器部件为1 1 1的点）到超级汇点的流量为无穷。
输出路径：1+n…n+n的点到1..n的点如果机器人部件都一样，如果流量变低了。代表有机器人转换，这时候记录那个点到那个点，还有转换了多少，后面输出就好了。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int inf = 0x3f3f3f3f;
const int mm = 250;
struct node
{
    int to, w, cap, next;
};
node Map[50000];
int p;
int out[mm][15], in[mm][15], a[mm], vis[mm], head[mm], cnt;
int dfs(int s, int t, int f)
{
    if(s == t) return f;
    vis[s] = 1;
    for(int i = head[s]; ~i; i = Map[i].next)
    {
        int to = Map[i].to, &w = Map[i].w;
        if(!vis[to] && w)
        {
            int d = dfs(to, t, min(f, w));
            if(d > 0)
            {
                w -= d;
                Map[i^1].w += d;
                return d;
            }
        }
    }
    return 0;
}
int Ek(int s, int t)
{
    int Max_flow = 0;
    for(;;){
        memset(vis, 0, sizeof(vis));
        int d = dfs(s, t, inf);
        if(!d) break;
        Max_flow += d;
    }
    return Max_flow;
}
bool check(int u, int v)
{
    for(int i = 1; i <= p; i++)
    {
        if(in[v][i] == 2) continue;
        if(out[u][i] != in[v][i]) return 0;
    }
    return 1;
}
void add(int u, int v, int w)//前向星存图
{
    Map[cnt].to = v;
    Map[cnt].w = w;
    Map[cnt].cap = w;
    Map[cnt].next = head[u];
    head[u] = cnt++;
    Map[cnt].to = u;
    Map[cnt].w = 0;
    Map[cnt].cap = 0;
    Map[cnt].next = head[v];
    head[v] = cnt++;
}
int main()
{
    int i, j, n;
    while(~scanf("%d %d", &p, &n))
    {
        memset(out, 0, sizeof(out));//用来存i这个转换方式转换后的部件
        memset(in, 0, sizeof(in));//用来存i这个转换方式转换前的部件
        cnt = 0;
        memset(head, -1, sizeof(head));
        for(i = 1; i <= n; i++)//输入
        {
            scanf("%d", &a[i]);
            for(j = 1; j <= p; j++)
                scanf("%d", &in[i][j]);
            for(j = 1; j <= p; j++)
                scanf("%d", &out[i][j]);
        }
        a[0] = a[2 * n + 1] = inf;//超级源点，超级汇点
        //超级源点的部件就是0 0 0
        for(i = 1; i <= p; i++)//超级汇点的部件就是1 1 1
            in[2 * n + 1][i] = 1;
        for(i = 1; i <= n; i++)//i点到i + n点流量a[i]
            add(i, i + n, a[i]);
        for(i = 1; i <= n; i++)
        {
            for(j = 1; j <= n; j++)
            {
                if(i != j)
                {
                    if(check(i, j))//判断out[i]是不是和in[j]一样
                    {
                        add(i + n, j, inf);
                    }
                }
            }
        }
        for(i = 1; i <= n; i++)//超级源点到1-n。i+n到超级汇点
        {
            if(check(0, i)) add(0, i, inf);
            if(check(i, 2 * n + 1)) add(i + n, 2 * n + 1, inf);
        }
        //求最大流
        int Max_flow = Ek(0, 2 * n + 1);
        printf("%d", Max_flow);
        int ans[10000][4], top = 0;
        for(i = 1; i <= n; i++)//保存路径
        {
            for(j = head[i + n]; ~j; j = Map[j].next)
            {
                if(Map[j].to == (2 * n + 1)) continue;
                int w = Map[j].w, cap = Map[j].cap, to = Map[j].to;
                if(w < cap)//代表有转换，记录路径
                {
                    ans[top][0] = i;
                    ans[top][1] = to;
                    ans[top++][2] = cap - w;
                }
            }
        }
        printf(" %d\n", top);//输出路径
        for(i = 0; i < top; i++)
        {
            printf("%d %d %d\n", ans[i][0], ans[i][1], ans[i][2]);
        }
    }
    return 0;
}