POJ3281

題目鏈接:http://poj.org/problem?id=3281

題目大意:

有F中食物D種飲料,每種食物或飲料只能分配給一頭牛。最多能有多少頭牛可以同時得到喜歡的食物和飲料

思路:

因爲要使每頭牛鬥要有食物和飲料,而且爲了分給儘可能多的牛,所以也就是一頭牛最好分得一種食物和飲料,在牛和牛之間要建立一個傳遞,使得容量爲1 ,也就變成要求食物爲1了,在進入匯點時也要對應成每種飲料一條邊但是可能會出現多種食物對應一頭牛得情況,所以需要將邊的鏈接方式變成s->食物->牛1->牛2->飲料->t使他們的容量都爲1,用最大流Dinic進行求解,否則會TLE

代碼:

 

<span style="font-size:18px;"><span style="font-size:18px;">#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn = 110;//牛
const int maxf = 110;///食物
const int maxd = 110;///飲料
const int inf = 2;
struct edge{int to, cap, rev; };
vector<edge> G[maxn*2+maxf+maxd];
int level[maxn*2+maxf+maxd];
int iter[maxn*2+maxf+maxd];
bool liked[maxn][maxd];
bool likef[maxn][maxf];
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 u, int t, int f)
{
    if(u == t)
        return f;
    for(int &i = iter[u]; i < G[u].size(); i++)
    {
        edge &e = G[u][i];
        if( e.cap > 0 && level[u] <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 main()
{
    int n,F,D;
    scanf("%d%d%d", &n, &F, &D);
        memset(likef, false, sizeof(likef));
        memset(liked, false, sizeof(liked));
        int d,f,di,fi;
        for(int i = 0; i < n; i++)
        {
            scanf("%d%d", &f, &d);
            for(int j = 0; j < f; j++)
            {
                scanf("%d",&di);
                likef[i][di-1] = true;
            }
            for(int j = 0; j < d; j++)
            {
                scanf("%d", &fi);
                liked[i][fi-1] = true;
            }
        }
        int s = n*2+F+D, t = s+1;
        for(int i = 0; i< F; i++)///在s與食物之間連線
        {
            add_edge(s, 2*n+i, 1);
        }
        for(int i = 0; i< D; i++)
        {
            add_edge(2*n+F+i, t, 1);
        }
        for(int i = 0; i < n; i++)
        {
            add_edge(i, n+i, 1);
            for(int j = 0; j < F; j++)
            {
                if(likef[i][j])
                    add_edge(2*n+j, i, 1);
            }
            for(int j = 0; j < D; j++)
            {
                if(liked[i][j])
                    add_edge(n+i, 2*n+F+j, 1);
            }
        }
        printf("%d\n", max_flow(s,t));
    return 0;

}
</span></span>


 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章