POJ 3436ACM Computer Factory 网络流

POJ 3436ACM Computer Factory  网络流

一、题意

ACM的电脑由p个部分组成，只有p个部分都存在才能正常工作，现有n台机器，每台机器有三个参数，即：生产效率q，输入标准，输出标准，为了提高生产效率，对这些机器进行重新组网，求最大的生产效率。

二、解题思路

这题可以这样建图：(1)将每台机器看成两个点，入点和出点，对于每台机器的生产效率可以看作，入点和出点之间有一条容量（cap）为q的变。
（3）分别设置源点s，和汇点t，若某机器的输入标准中能接受初始状态（即：没有任何配件的状态）则s和该机器的入点有一条INF的边，同理，若某机器的输出为可接受状态（即：全为1的状态），则，该机器的出点与汇点t有一条INF的边。

（2）枚举两台不同机器的有序组合（i，j），若i的输出标准满足j的输入标准，则在i的出点与j的入点之间建一条cap为无穷大（INF）的边。

图建好后，s-t的最大流即为所求，用dinic算法进行处理。这一题要求输出答案的边的流量，用dinic算法处理后的残余网络，与原图进行对比即可。

三、代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<map>
#include<string>
#include<queue>
#include<vector>
#include<list>
//#pragma comment(linker,"/STACK:1024000000,1024000000")
using namespace std;
const int MAX_V=205;
#define INF 0x3f3f3f3f
struct edge{int to,cap,rev;};
vector<edge>G[MAX_V];
int level[MAX_V];
int iter[MAX_V];
void add_edge(int from,int to,int cap)
{
    G[from].push_back((edge){to,cap,G[to].size()});
    G[to].push_back((edge){from,0,G[from].size()-1});
}
void bfs(int s)
{
    memset(level,-1,sizeof level);
    queue<int> que;
    level[s]=0;
    que.push(s);
    while(!que.empty())
    {
        int v=que.front();que.pop();
        for(int i=0;i<G[v].size();i++)
        {
            edge &e=G[v][i];
            if(e.cap>0&&level[e.to]<0)
            {
                level[e.to]=level[v]+1;
                que.push(e.to);
            }
        }
    }
}
int dfs(int v,int t,int f)
{
    if(v==t) return f;
    for(int &i=iter[v];i<G[v].size();i++)
    {
        edge& e=G[v][i];
        if(e.cap>0&&level[v]<level[e.to])
        {
            int d=dfs(e.to,t,min(f,e.cap));
            if(d>0)
            {
                e.cap-=d;
                G[e.to][e.rev].cap+=d;
                return d;
            }
        }
    }
    return 0;
}
int max_flow(int s,int t)
{
    int flow=0;
    for(;;)
    {
        bfs(s);
        if(level[t]<0) return flow;
        memset(iter,0,sizeof iter);
        int f;
        while((f=dfs(s,t,INF))>0)
        {
            flow+=f;
        }
    }
}
int p,n;
int mac[55][15][2],cap[55];
vector<edge> res;
int main()
{
    while(~scanf("%d %d",&p,&n)&&p+n)
    {
        res.clear();
        for(int i=0;i<MAX_V;i++) G[i].clear();
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&cap[i]);
            for(int j=1;j<=p;j++) scanf("%d",&mac[i][j][0]);
            for(int j=1;j<=p;j++) scanf("%d",&mac[i][j][1]);
            add_edge(2*i-1,2*i,cap[i]);
            int flag=0;for(int j=1;j<=p;j++) if(mac[i][j][0]==1) {flag=1;break;} if(!flag) add_edge(0,2*i-1,INF);
            flag=0;for(int j=1;j<=p;j++) if(mac[i][j][1]==0) {flag=1;break;} if(!flag) add_edge(2*i,2*n+1,INF);
        }
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
        {
            if(i==j) continue;
            int flag=1;
            for(int k=1;k<=p;k++)
            {
                if(mac[i][k][1]==0&&mac[j][k][0]==1) {flag=0;break;}
                if(mac[i][k][1]==1&&mac[j][k][0]==0) {flag=0;break;}
            }
            if(flag) add_edge(2*i,2*j-1,INF);
        }
        int ans=max_flow(0,2*n+1);
        for(int i=2;i<=2*n;i+=2)
        {
            for(int j=0;j<G[i].size();j++)
            {
                edge e=G[i][j];
                if(e.to!=2*n+1&&e.to!=i-1&&e.cap!=INF) res.push_back((edge){i/2,e.to/2+1,INF-e.cap});
            }
        }
        printf("%d %d\n",ans,res.size());
        for(int i=0;i<res.size();i++)
        {
            printf("%d %d %d\n",res[i].to,res[i].cap,res[i].rev);
        }
    }
    return 0;
}