luogu1402 酒店之王

题目描述

XX酒店的老板想成为酒店之王，本着这种希望，第一步要将酒店变得人性化。由于很多来住店的旅客有自己喜好的房间色调、阳光等，也有自己所爱的菜，但是该酒店只有p间房间，一天只有固定的q道不同的菜。

有一天来了n个客人，每个客人说出了自己喜欢哪些房间，喜欢哪道菜。但是很不幸，可能做不到让所有顾客满意（满意的条件是住进喜欢的房间，吃到喜欢的菜）。

这里要怎么分配，能使最多顾客满意呢？

输入输出格式

输入格式：

第一行给出三个正整数表示n,p,q(<=100)。

之后n行，每行p个数包含0或1，第i个数表示喜不喜欢第i个房间（1表示喜欢，0表示不喜欢）。

之后n行，每行q个数，表示喜不喜欢第i道菜。

输出格式：

最大的顾客满意数。

输入输出样例

输入样例#1：

2 2 2
1 0
1 0
1 1
1 1

输出样例#1：

网络流，直接建图，喜欢的连边，容量为1
注意一个人要拆成两个点，如果不拆，那么这个人会被多组酒店和菜匹配

# include <bits/stdc++.h>

# define IL inline

# define RG register

# define Fill(a, b) memset(a, b, sizeof(a))

# define Copy(a, b) memcpy(a, b, sizeof(a))

using namespace std;

typedef long long ll;

const int _(410), __(4e4 + 10), INF(2147483647);

IL ll Read(){

    RG char c = getchar(); RG ll x = 0, z = 1;

    for(; c < '0' || c > '9'; c = getchar()) z = c == '-' ? -1 : 1;

    for(; c >= '0' && c <= '9'; c = getchar()) x = (x << 1) + (x << 3) + (c ^ 48);

    return x * z;

}

int n, p, q, w[__], fst[_], nxt[__], to[__], cnt;

int S, T, lev[_], cur[_], max_flow, ans;

queue <int> Q;

IL void Add(RG int u, RG int v, RG int f){

    w[cnt] = f; to[cnt] = v; nxt[cnt] = fst[u]; fst[u] = cnt++;

    w[cnt] = 0; to[cnt] = u; nxt[cnt] = fst[v]; fst[v] = cnt++;

}

IL int Dfs(RG int u, RG int maxf){

    if(u == T) return maxf;

    RG int ret = 0;

    for(RG int &e = cur[u]; e != -1; e = nxt[e]){

        if(lev[to[e]] != lev[u] + 1 || !w[e]) continue;

        RG int f = Dfs(to[e], min(w[e], maxf - ret));

        ret += f; w[e ^ 1] += f; w[e] -= f;

        if(ret == maxf) break;

    }

    return ret;

}

IL bool Bfs(){

    Fill(lev, 0); lev[S] = 1; Q.push(S);

    while(!Q.empty()){

        RG int u = Q.front(); Q.pop();

        for(RG int e = fst[u]; e != -1; e = nxt[e]){

            if(lev[to[e]] || !w[e]) continue;

            lev[to[e]] = lev[u] + 1;

            Q.push(to[e]);

        }

    }

    return lev[T];

}

int main(RG int argc, RG char* argv[]){

    n = Read(); p = Read(); q = Read(); Fill(fst, -1); T = 2 * n + p + q + 1;

    for(RG int i = 1; i <= p; i++) Add(S, i, 1);

    for(RG int i = 1; i <= q; i++) Add(i + 2 * n + p, T, 1);

    for(RG int i = 1; i <= n; i++) Add(i + p, i + p + n, 1);

    for(RG int i = 1; i <= n; i++)

        for(RG int j = 1; j <= p; j++)

            if(Read()) Add(j, i + p, 1);

    for(RG int i = 1; i <= n; i++)

        for(RG int j = 1; j <= q; j++)

            if(Read()) Add(i + n + p, j + 2 * n + p, 1);

    while(Bfs()) Copy(cur, fst), max_flow += Dfs(S, INF);

    printf("%d\n", max_flow);

    return 0;

}

秒客网