网络流三·二分图多重匹配 HihoCoder - 1393

描述

学校的秋季运动会即将开始，为了决定参赛人员，各个班又开始忙碌起来。

小Hi和小Ho作为班上的班干部，统计分配比赛选手的重任也自然交到了他们手上。

已知小Hi和小Ho所在的班级一共有N名学生(包含小Hi和小Ho)，编号依次为1..N。

运动会一共有M项不同的比赛，编号为1..M。第i项比赛每个班需要派出m[i]名选手参加。

根据小Hi和小Ho的统计，编号为i的学生表示最多同时参加a[i]项比赛，并且给出他所擅长的b[i]项比赛的编号。

小Hi和小Ho希望将每个学生都安排到他所擅长的比赛项目，以增加夺冠的可能性。同时又要考虑满足每项比赛对人数的要求，当然给一个学生安排的比赛项目也不能超过他愿意参加的比赛项目数量。

根据统计的结果，小Hi和小Ho想知道能否有一个合适的安排，同时满足这些条件。

提示：二分图多重匹配

输入

第1行：1个整数T，表示一共有T(2≤T≤5)组数据，每组数据按如下格式给出：

第1行：2个正整数N,M。1≤N≤100，1≤M≤100。

第2行：M个整数，第i个数表示第i个项目需要的选手数量m[i]。1≤m[i]≤N。

第3..N+2行：若干整数，第i+2行表示编号为i的学生的信息。先是a[i],b[i]，接下来b[i]个整数，表示其所擅长的b[i]个项目。1≤a[i]≤M

输出

第1..T行：第i行表示第i组数据能否满足要求，若能够输出”Yes”，否则输出”No”。

Sample Input
2
4 3
1 2 2
1 2 1 2
2 2 1 3
1 1 2
1 2 2 3
4 3
2 2 2
1 2 1 2
2 2 1 3
1 1 2
1 2 2 3
Sample Output
Yes
No
发现hihoCoder是一个不错的学习算法的网站，用网络流解决的算法，这个思想在hiho的解释讲的非常清楚。

#include<cstdio>
#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
int n,m;
struct node1
{
    int to;
    int cap;
}edge[30000];
int k;
int total;
vector<int> graph[205];
void addEdge(int x,int y,int c)
{
    edge[k].to=y;
    edge[k].cap=c;
    graph[x].push_back(k);
    k++;
    edge[k].to=x;
    edge[k].cap=0;
    graph[y].push_back(k);
    k++;
}
struct node
{
    int now;
    int lastIndex;
    int index;
}que[30000];
int head,tail;
bool vis[400];
bool work()
{
    bool sign;
    int ans=0;
    node start;
    node endd;
    for(;;)
    {
        //cout<<"haha"<<endl;
        memset(vis,false,sizeof(vis));
        start.lastIndex=-1;
        start.now=0;
        head=tail=0;
        que[tail++]=start;
        sign=true;
        vis[0]=true;
        while(head<tail&&sign)
        {
            node temp=que[head++];
            int u=temp.now;
            for(int i=0;i<graph[u].size();i++)
            {
                int index=graph[u][i];
                int v=edge[graph[u][i]].to;
                if(vis[v]||edge[index].cap==0)
                    continue;
                if(v==n+m+1)
                {
                    endd.index=index;
                    endd.lastIndex=head-1;
                    sign=false;
                    break;
                }
                else
                {
                    vis[v]=true;
                    node out;
                    out.now=v;
                    out.lastIndex=head-1;
                    out.index=index;
                    que[tail++]=out;
                }
            }
        }
        if(sign)
            break;
        else
        {
            ans++;
            while(endd.lastIndex!=-1)
            {
                int index=endd.index;
                edge[index].cap--;
                edge[index^1].cap++;
                endd=que[endd.lastIndex];
            }
        }
    }
    //cout<<ans<<endl;
    //cout<<total<<endl;
    return ans==total;
}
int main()
{
   int t;
   scanf("%d",&t);
   int a,b;
   while(t--)
   {
       scanf("%d%d",&n,&m);
       for(int i=0;i<=n+m+1;i++)
            graph[i].clear();
       k=0;
       int p=n+m+1;
       total=0;
       for(int i=1;i<=m;i++)
       {
           scanf("%d",&b);
           addEdge(n+i,p,b);
           total+=b;
       }
       for(int i=1;i<=n;i++)
       {
           scanf("%d%d",&a,&b);
           addEdge(0,i,a);
           while(b--)
           {
               scanf("%d",&a);
               addEdge(i,n+a,1);
           }
       }
       if(work())
        printf("Yes\n");
       else
        printf("No\n");
   }
    return 0;
}