poj1149 经典建模

http://wenku.baidu.com/view/0ad00abec77da26925c5b01c.html 以上内容均为转载

#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=;
const int INF=;
vector<int>G[];
bool mp[N][N],used[];
int tot,head[N],dis[N],st,ed,val[],sum[N];
struct node{
   int to,next,w;
}e[N*N*];
void init(){
   tot=;
   memset(head,-,sizeof(head));
}
void add(int u,int v,int w){
   e[tot].to=v;
   e[tot].w=w;
   e[tot].next=head[u];
   head[u]=tot++;
}
bool bfs(){
   memset(dis,-,sizeof(dis));
   queue<int>Q;
   Q.push(st);
   dis[st]=;
   while(!Q.empty()){
    int u=Q.front();
    Q.pop();
    for(int i=head[u];i+;i=e[i].next){
        int v=e[i].to;
        if(dis[v]==-&&e[i].w>){
            dis[v]=dis[u]+;
            if(v==ed) return true;
            Q.push(v);
        }
    }
   }
   return false;
}
int dfs(int s,int low){
   if(s==ed||low==) return low;
   int ans=low,a;
   for(int i=head[s];i+;i=e[i].next){
    int v=e[i].to;
    if(dis[v]==dis[s]+&&e[i].w>&&(a=dfs(v,min(ans,e[i].w)))){
        ans-=a;
        e[i].w-=a;
        e[i^].w+=a;
        if(!ans) return low;
    }
   }
   if(low==ans) dis[s]=-;
   return low-ans;
}
int main(){
    int n,m,x,num;
    while(scanf("%d%d",&n,&m)!=EOF){
        st=;ed=m+;
        memset(sum,,sizeof(sum));
        for(int i=;i<=n;++i) G[i].clear();
        for(int i=;i<=n;++i) scanf("%d",val+i);
        memset(used,,sizeof(used));
        init();
        for(int i=;i<=m;++i) {
            scanf("%d",&num);
            while(num--){
                scanf("%d",&x);
                if(!used[x]) {used[x]=;sum[i]+=val[x];}
                G[x].push_back(i);
            }
            scanf("%d",&x);
            add(i,ed,x);
            add(ed,i,);
        }
        memset(mp,,sizeof(mp));
        for(int i=;i<=n;++i){
            for(int j=;j<(int)G[i].size();++j){
                if(!j&&!mp[][G[i][j]]) {
                    mp[G[i][j]][]=mp[][G[i][j]]=;
                    add(,G[i][j],sum[G[i][j]]);
                    add(G[i][j],,);
                }
                else if(!mp[G[i][]][G[i][j]]) {
                    add(G[i][],G[i][j],INF);
                    add(G[i][j],G[i][],);
                    mp[G[i][]][G[i][j]]=;
                }
            }
        }
        int ans=;
        while(bfs()) ans+=dfs(,INF);
        printf("%d\n",ans);
    }
}