1. 程式人生 > >poj 3281 Dining

poj 3281 Dining

題意:   有N頭牛F種食物可以製作,D種飲料可以製作   然後每行代表一頭牛的喜好,開頭兩個數fi,di表示這頭牛喜歡fi種食物,di種飲料,接下來fi個數表示喜歡的食物編號,di個數表示喜歡的飲料的編號   現在主人使用最優決策製作出F種食物和D種飲料,問怎麼喂才能使盡可能多的牛餵飽(餵飽=一份食物一份飲料,且一頭牛最多消耗一份食物和一份飲料)  輸出最多餵飽的牛數。

 

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>

using namespace std;

#define INF 0x3f3f3f3f

const int maxn=500;

int N,F,D;
int c[maxn][maxn],dep[maxn];

int bfs(int s,int t)
{
    queue<int > q;
    while(!q.empty()) q.pop();

    memset(dep,-1,sizeof(dep));

    dep[s]=0;
    q.push(s);

    while(!q.empty())
    {
        int u=q.front();
        q.pop();

        for(int v=1;v<=t;v++)
        {
            if(c[u][v]>0&&dep[v]==-1){
                dep[v]=dep[u]+1;
                q.push(v);
            }
        }
    }
    return dep[t]!=-1;
}

int dfs(int u,int mi,int t)
{
    if(u==t) return mi;
    int tmp;

    for(int v=1;v<=t;v++)
    {
        if(c[u][v]>0&&dep[v]==dep[u]+1&&(tmp=dfs(v,min(mi,c[u][v]),t))){

            c[u][v]-=tmp;
            c[v][u]+=tmp;
            return tmp;
        }
    }
    return 0;
}

int dinic(int start,int ends)
{
    int ans=0,tmp;

    while(bfs(start,ends))
    {
        while(1){
            tmp=dfs(start,INF,ends);
            if(tmp==0) break;
            ans+=tmp;
        }
    }

    return ans;
}
int main()
{

    while(~scanf("%d%d%d",&N,&F,&D))
    {
        memset(c,0,sizeof(c));

       int start=1,ends=1+F+2*N+D+1;

        for(int i=2;i<=F+1;i++) ///建立源點到食物的邊
            c[1][i]=1;

        for(int i=1+F+1;i<=1+F+N;i++) ///建立牛與牛之間的邊
            c[i][i+N]=1;

        for(int i=1+F+2*N+1;i<=1+F+2*N+D;i++) ///建立飲料到終點的邊
            c[i][ends]=1;


        for(int i=1;i<=N;i++)
        {
            int food,drink;
            scanf("%d%d",&food,&drink);

            int item;
            while(food--)
            {
                scanf("%d",&item);

                c[item+1][1+F+i]=1; ///建立食物與牛之間的邊
            }
            while(drink--)
            {
                scanf("%d",&item);

                c[1+F+N+i][1+F+2*N+item]=1; ///建立牛與飲料之間的邊
            }
        }

        printf("%d\n",dinic(start,ends));
    }

    return 0;
}