[hihocoder1393] 二分图多重匹配

题意：

二分图多重匹配
问你匹配完之后，到汇点是否满流

题解：

最大流

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define ll long long
#define N 100010 
using namespace std;

int T,n,m,t,s,sum,inf=1<<30,e_num=0;
int nxt[N],to[N],w[N],h[N],lev[N],cur[N];

int gi() {
  int x=0,o=1; char ch=getchar();
  while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
  if(ch=='-') o=-1,ch=getchar();
  while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
  return o*x;
}

void add(int x, int y, int z) {
  nxt[e_num]=h[x],to[e_num]=y,w[e_num]=z,h[x]=e_num++;
}

bool bfs() {
  queue<int> q;//2:清空队列 
  memset(lev,0,sizeof(lev));
  q.push(s),lev[s]=1;//1:初始值位1 
  while(!q.empty()) {
    int u=q.front();
    q.pop();
    for(int i=h[u]; i!=-1; i=nxt[i]) {
      int v=to[i];
      if(!lev[v] && w[i]) {
		lev[v]=lev[u]+1;
		q.push(v);
		if(v==t) return true;
      }
    }
  }
  return false;
}

int dfs(int u, int f) {
  if(u==t) return f;
  int tag=0,c;
  for(int &i=cur[u]; i!=-1; i=nxt[i]) {
    int v=to[i];
    if(lev[v]==lev[u]+1 && w[i]) {
      c=dfs(v,min(f-tag,w[i]));
      w[i]-=c,w[i^1]+=c,tag+=c;
      if(tag==f) return tag;
    }
  }
  return tag;
}

bool dinic() {
  int ans=0;
  while(bfs()) {
    for(int i=0; i<=t; i++) cur[i]=h[i];
    ans+=dfs(s,inf);   
  }
  if(ans==sum) return true;
  else return false;
}

int main() {
  int T=gi();
  while(T--) {
    memset(h,-1,sizeof(h));
    n=gi(),m=gi(),t=n+m+1,sum=e_num=0;
    for(int i=1; i<=m; i++) {
      int x=gi();
      add(i+n,t,x),add(t,i+n,0);
      sum+=x;
    }
    for(int i=1; i<=n; i++) {
      int x=gi(),y=gi(),z;
      add(0,i,x),add(i,0,0);
      while(y--) {
		z=gi();
		add(i,z+n,1),add(z+n,i,0);
      }
    }
    if(dinic()) puts("Yes");
    else puts("No");
  }
  return 0;
}