luogu P2417 课程

题目描述

n个学生去p个课堂,每一个学生都有自己的课堂,并且每个学生只能去一个课堂,题目要求能够安排每一个课堂都有人吗?

输入格式

第一行是测试数据的个数,

每组测试数据的开始分别是p和n,

接着p行,每行的开始是这个课堂的学生人数m,接着m个数代表该课堂的学生编号

输出格式

如果该组数据能够这样安排就输出YES,否则输出NO。

说明/提示

对于100%的数据,n\le 100,m\le 20000n≤100,m≤20000

网络流板子

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
const int inf=1<<29,N=6e4+10,M=3e5+10;
using namespace std;
int next[M],head[N],go[M],edge[M],tot;
int d[N],maxflow;
void add(int u,int v,int o){
    next[++tot]=head[u];head[u]=tot;go[tot]=v;edge[tot]=o;
    next[++tot]=head[v];head[v]=tot;go[tot]=u;edge[tot]=0;
}
int n,m,s,t;
bool bfs(){
    memset(d,0,sizeof(d));
    queue<int>q;
    q.push(s);d[s]=1;
    while(q.size()){
        int x=q.front();q.pop();
        for(int i=head[x];i;i=next[i])
        if(edge[i]&&!d[go[i]]){
            q.push(go[i]);
            d[go[i]]=d[x]+1;
            if(go[i]==t)return 1;
        }
    }
    return 0;
}
int dinic(int x,int flow){
    if(x==t)return flow;
    int rest=flow,k;
    for(int i=head[x];i&&rest;i=next[i])
    if(edge[i]&&d[go[i]]==d[x]+1){
        k=dinic(go[i],min(rest,edge[i]));
        if(!k)d[go[i]]=0;
        edge[i]-=k;
        edge[i^1]+=k;
        rest-=k;
    }
    return flow-rest;
}
int main(){
    int T;
    cin>>T;
    while(T--){
        cin>>n>>m;tot=1;
        s=0,t=n+m+1;
        swap(n,m);
        for(int i=1,p;i<=m;i++){
            scanf("%d",&p);
            add(i+n,t,1);
            for(int j=1,x;j<=p;j++){
                scanf("%d",&x);
                add(x,i+n,1);
            }
        }
        for(int i=1;i<=n;i++)add(s,i,1);
        
        int flow=0;
        maxflow=0;
        while(bfs())
        while(flow=dinic(s,inf))maxflow+=flow;
        if(maxflow==m)printf("YES\n");
        else printf("NO\n");
        memset(next,0,sizeof(next));
        memset(head,0,sizeof(head));
        memset(go,0,sizeof(go));
    }
    

}

猜你喜欢

转载自www.cnblogs.com/naruto-mzx/p/11624627.html