题目链接
https://www.luogu.org/problemnew/show/P2763
试题库问题
可以发现这道题有点类似二分图匹配,唯一不同的是每个类型的数量有限制,我们回顾一下用最大流时的做法,设两边集合为A,B,S向A连流量为1的边,A向B连流量为1的边,B向T连流量为1的边,这样保证了每对AB只会产生一的贡献,那么我们这里所有B(表示类型)向T连的边的流量改为该题目联系要求的数量,其他不变,那么每条S-A-B-T的增广路表示的就是A这道题选B类型,因为S-A的流量为1,就保证了每个A只能流到一个B,B-T则表示这个类型的题目我只需要这么多,那么最后最大流流完判一下maxflow是不是于要求的题目数量相同,如果相同说明是可以做到的,否则No Solution!这个非常显然,至于输出方案,每个A看他这个流量流向哪个B,他就属于哪个类型的题目
#include<cstdio>
#include<algorithm>
#include<cctype>
#include<cstring>
#include<iostream>
#include<vector>
#define LL long long
#define INF (2139062143)
#define N (1000001)
using namespace std;
int tot,k,n,x,S,T,y,all,sum,ans;
int a[N],fl[N],nxt[N],head[N],h[N],cou[N],cur[N];
vector <int> q[N];
template <typename T> void read(T&t) {
t=0;
bool fl=true;
char p=getchar();
while (!isdigit(p)) {
if (p=='-') fl=false;
p=getchar();
}
do {
(t*=10)+=p-48;p=getchar();
}while (isdigit(p));
if (!fl) t=-t;
}
inline void add(int x,int y,int f){
a[++tot]=y,nxt[tot]=head[x],head[x]=tot,fl[tot]=f;
}
int dfs(int u,int maxflow){
if (u==T) return maxflow;
int used=0;
for (int p=cur[u];~p;p=nxt[p]){
cur[u]=p;
if (fl[p]>0&&h[a[p]]+1==h[u]){
int now=dfs(a[p],min(maxflow-used,fl[p]));
used+=now;
fl[p]-=now;
fl[p^1]+=now;
if (maxflow==used) return maxflow;
}
}
cur[u]=head[u];
if (--cou[h[u]]==0) h[S]=all;
h[u]++;
cou[h[u]]++;
return used;
}
int main(){
read(k),read(n);
S=k+n+1,T=k+n+2;
tot=1;
memset(head,-1,sizeof(head));
memset(cur,-1,sizeof(cur));
for (int i=1;i<=k;i++){
read(x);
sum+=x;
add(i+n,T,x);
add(T,i+n,0);
}
for (int i=1;i<=n;i++){
add(S,i,1);
add(i,S,0);
read(x);
while (x--){
read(y);
add(i,y+n,1);
add(y+n,i,0);
}
}
memset(h,0,sizeof(h));
all=n+k+2;
cou[0]=all;
while (h[S]<all) ans+=dfs(S,INF);
if (ans<sum){
puts("No Solution!");
return 0;
}
for (int i=1;i<=n;i++){
for (int p=head[i];p;p=nxt[p]){
if (fl[p]==0){
q[a[p]].push_back(i);
}
}
}
for (int i=1;i<=k;i++){
printf("%d: ",i);
for (int j=0;j<q[i+n].size();j++){
printf("%d ",q[i+n][j]);
}
puts("");
}
return 0;
}