洛谷P1402 酒店之王【网络流/点的限制/拆点】

题目链接：P1402 酒店之王

分析：对客人拆点加上限为1的限制；

#include
#define pb push_back
using namespace std;
typedef long long ll;
const int maxn=1200;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
struct node {int to,cap,rev;};
int dep[maxn],n,m,S,T;
vectorv[maxn];
void add(int from,int to,int cap)  //重边情况不影响
{
    v[from].pb({to,cap,(int)v[to].size()});
    v[to].pb({from,0,(int)v[from].size()-1});
}
int bfs(int s,int t)
{
    queueq;
    memset(dep,-1,sizeof(dep));
    dep[s]=0;
    q.push(s);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=0;i0&&dep[v[u][i].to]==-1)
            {
                dep[v[u][i].to]=dep[u]+1;
                q.push(v[u][i].to);
            }
    }
    return dep[t]!=-1;
}
int dfs(int u,int t,int micap)
{
    if(u==t)return micap;
    int tmp;
    for(int i=0;i0 && dep[u]+1==dep[now.to] && (tmp=dfs(now.to,t,min(now.cap,micap))))
        {
            now.cap-=tmp;
            v[now.to][now.rev].cap+=tmp;
            return tmp;
        }
    }
    return 0;
}
int dinic(int s,int t)
{
    int ans=0,tmp;
    while(bfs(s,t))
        while(1)
        {
            tmp=dfs(s,t,inf);
            if(tmp==0)break;
            ans+=tmp;
        }
    return ans;
}
int main()
{
    int n,p,q,x;scanf("%d%d%d",&n,&p,&q);
    int s=0,t=2*n+p+q+1;//memset(v,0,sizeof(v));
    for(int i=1;i<=p;i++) add(s,i,1);
    for(int i=1;i<=q;i++) add(i+p,t,1);
    for(int i=1;i<=n;i++) add(i+p+q,i+p+q+n,1);
    for(int i=1;i<=n;i++)for(int j=1;j<=p;j++) {scanf("%d",&x);if(x) add(j,i+p+q,1);} 
    for(int i=1;i<=n;i++)for(int j=1;j<=q;j++) {scanf("%d",&x);if(x) add(i+p+q+n,j+p,1);} 
    printf("%d\n",dinic(s,t));
}