给出
n
≤
1
e
3
n\leq1e3
n≤1e3个问题,然后每个问题有若干个类型,共有
k
≤
20
k\leq20
k≤20种类型,然后要组
m
m
m个题,给出每个类型的题数
t
i
t_i
ti,
∑
i
=
1
k
t
i
=
m
\sum_{i=1}^k t_i=m
∑i=1kti=m。求一种满足条件的方案。
源点向所有的题连
1
1
1的流量,题向它的类型也连
1
1
1的流量,它的每种类型向汇点连所需类型个数的流量。如果满流则说明方案有解。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
int sum=0;
vector<int> g[21];
namespace Dinic {
const int N=2e3+3,M=1e5+3;
int front[N],head[N],dep[N],tot=-1,n,m,S,T;
struct edge{ int v,nxt;ll w; }e[2*M];
void init() {
tot=-1;
for(int i=1;i<=n;i++) front[i]=-1;
}
void add(int u,int v,ll w) {
e[++tot].v=v,e[tot].w=w,e[tot].nxt=front[u],front[u]=tot;
e[++tot].v=u,e[tot].w=0,e[tot].nxt=front[v],front[v]=tot;
}
bool bfs() {
for(int i=1;i<=n;i++) dep[i]=0;
queue<int> q;
q.push(S);
dep[S]=1;
while(!q.empty()) {
int u=q.front();
q.pop();
for(int i=front[u];i!=-1;i=e[i].nxt) {
int v=e[i].v;
if(dep[v]==0&&e[i].w) {
dep[v]=dep[u]+1;
q.push(v);
}
}
}
return dep[T];
}
ll dfs(int cur,ll dis=1e18) {
if(cur==T) return dis;
ll flow=0,sum=0;
for(int &i=head[cur];i!=-1;i=e[i].nxt) {
int v=e[i].v;
if(dep[v]==dep[cur]+1&&e[i].w) {
flow=dfs(v,min(dis,e[i].w));
dis-=flow,sum+=flow;
e[i].w-=flow,e[i^1].w+=flow;
if(!dis) break;
}
}
return sum;
}
void solve() {
ll ans=0;
while(bfs()) {
for(int i=1;i<=n;i++) head[i]=front[i];
ans+=dfs(S);
}
if(ans!=sum) {
printf("No Solution!");
}
else {
for(int i=1;i<=n-m-2;i++) {
for(int j=front[i];j!=-1;j=e[j].nxt) {
if(e[j].v==S) continue;
if(e[j].w==0) {
g[e[j].v-(n-m-2)].push_back(i);
}
}
}
for(int i=1;i<=m;i++) {
printf("%d: ",i);
for(int j=0;j<g[i].size();j++) {
printf("%d%c",g[i][j],j+1==g[i].size()?'\n':' ');
}
}
}
}
}
int rec[22];
int main() {
int m,n;
scanf("%d%d",&m,&n);
for(int i=1;i<=m;i++) {
scanf("%d",&rec[i]);
sum+=rec[i];
}
Dinic::n=n+m+2;
Dinic::m=m;
Dinic::init();
int S=n+m+1;
int T=n+m+2;
Dinic::S=S;
Dinic::T=T;
for(int tot,u=1;u<=n;u++) {
scanf("%d",&tot);
for(int v,j=1;j<=tot;j++) {
scanf("%d",&v);v+=n;
Dinic::add(u,v,1);
}
}
for(int i=1;i<=n;i++)
Dinic::add(S,i,1);
for(int i=n+1;i<=n+m;i++)
Dinic::add(i,T,rec[i-n]);
Dinic::solve();
return 0;
}