POJ1149 PIGS 解题报告

 网络流刷的第一题，从早上8:00的通识课开始想这题。最终在晚上才AC了……网络流好难

/*
Run ID    User    Problem    Result    Memory    Time    Language    Code Length    Submit Time
13647887    yaodongen    1149    Accepted    260K    0MS    C++    3034B    2014-11-20 23:45:03
*/
#include <iostream>
#include <sstream>
#include <ios>
#include <iomanip>
#include <functional>
#include <algorithm>
#include <vector>
#include <string>
#include <list>
#include <queue>
#include <deque>
#include <stack>
#include <set>
#include <map>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <climits>
#include <cctype>
#define INF 0x3f3f3f3f
#define MP(X,Y) make_pair(X,Y)
#define PB(X) push_back(X)
#define REP(X,N) for(int X=0;X<N;X++)
#define REP2(X,L,R) for(int X=L;X<=R;X++)
#define DEP(X,R,L) for(int X=R;X>=L;X--)
#define CLR(A,X) memset(A,X,sizeof(A))
#define IT iterator
#define PI 3.14159265358979323846
#define _ ios_base::sync_with_stdio(0);cin.tie(0);
#define X first
#define Y second
#define MAX_V 10101
#define maxn 110
using namespace std;
typedef long long ll;
typedef pair<int,int>PII;

struct edge{
    int to,cap,rev;
    edge(int a,int b,int c):to(a),cap(b),rev(c){}
};
vector<edge> G[maxn];
int level[maxn];
int iter[maxn];

void add_edge(int from ,int to,int cap){
    G[from].PB(edge(to,cap,G[to].size()));
    G[to].PB(edge(from,0,G[from].size()-1));
}
void bfs(int s){
    memset(level,-1,sizeof(level));
    level[s]=0;
    queue<int >que;
    que.push(s);
    while(!que.empty()){
        int u=que.front();que.pop();
        for(int i=0;i<G[u].size();++i){
            edge &e=G[u][i];
            if(e.cap>0 && level[e.to]<0){
                level[e.to]=level[u]+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[e.to]>level[v]){
            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;
    while(1){
        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 A[1010];//猪栏中猪 
int own[1010];
bool used[maxn][maxn];
int main()
{
    //猪和顾客 
    int n,m;
    while(cin>>n>>m){
        //超级源点 超级汇点 
        int s=0,t=100+1;
        for(int i=0;i<maxn;++i)
            G[i].clear();
        for(int i=1;i<=n;++i){
            cin>>A[i];//猪栏i中猪的个数 
        }
        //最近开启猪栏的人
        memset(own,-1,sizeof(own)); 
        //是否已经建立边 
        memset(used,0,sizeof(used)); 
        for(int i=1;i<=m;++i){//第几个顾客 
            int tt;
            cin>>tt;//钥匙数 
            int temp;//猪栏的序号 
            for(int j=0;j<tt;++j){
                cin>>temp;
                if(own[temp]<0){//猪栏从未被打开过 
                    own[temp]=i;
                    //已经建立边，直接容量增加，不去新建边 
                    if(used[0][i]){
                        for(int k=G[0].size()-1;k>=0;--k){
                            if(G[0][k].to==i){
                                G[0][k].cap+=A[temp];
                                break;
                            }
                        }
                    }
                    else{
                        add_edge(0,i,A[temp]);
                        used[0][i]=1;
                    }
                }
                else{
                    int q=own[temp];
                    if(!used[q][i]){
                        add_edge(q,i,INF);
                        used[q][i]=1;
                    }
                    own[temp]=i;
                }
            }
            cin>>temp;
            add_edge(i,t,temp);
        }
        cout<<max_flow(s,t)<<endl;
    }
    return 0;
}