首先表示这道题有点坑,可能AC代码测试样例结果不一样。。。。
首先题目里表示需要 k 个题型 n 道题目
那么这很先看上去就是个二分图匹配。。。但是我用的网络流做法。。。
把每个题型 和 终点 建边 输入 容量
把每个题目 和起点建边 由于只能选一道 容量为 1
之后 我们把题目和可对应的题型之间建边 容量也设为 1
跑一下最大流就可算出最多的匹配方案 如果不等于 m就无解否则有解
但是题目里要求输出对应的题目,只要确认 某道题目和某个题型之间的边 如果被割了, 很显然 这道题目是被选中在 这个题型里了(注意不要看反向边)可以记录个连通图,最后根据连通情况输出就完事了。
以下是 AC 代码
#include <iostream>
#include <cstdio>
#include <queue>
#include <cmath>
#include <cstring>
using namespace std;
const int maxn=1105;
const int maxe=60007;
const int inf=0x3f3f3f3f;
using namespace std;
struct edge
{
int to,next,val;
}g[maxe];
int x,to,cnt,n,m,k;
int s,t,ans;
int head[maxn],dis[maxn];
int f[25][maxn];
void add(int x,int to,int w)
{
g[++cnt].to=to; g[cnt].val=w; g[cnt].next=head[x]; head[x]=cnt;
g[++cnt].to=x; g[cnt].val=0; g[cnt].next=head[to]; head[to]=cnt;
}
bool bfs()
{
memset(dis,0,sizeof(dis));
dis[s]=1;
queue <int> q;
q.push(s);
while(q.size())
{
int u=q.front();
q.pop();
for (int i=head[u];i>0;i=g[i].next)
{
int v=g[i].to;
if (dis[v]==0 && g[i].val)
{
dis[v]=dis[u]+1;
if (v==t)
return true;
q.push(v);
}
}
}
return false;
}
int dfs(int x,int maxf)
{
if ((x==t) || (maxf==0))
return maxf;
int ret=0;
for (int i=head[x];i>0;i=g[i].next)
{
int to=g[i].to;
if ((g[i].val) && (dis[x]+1==dis[to]))
{
int flow=dfs(to,min(g[i].val,maxf-ret));
g[i].val-=flow;
g[i^1].val+=flow;
ret+=flow;
}
}
return ret;
}
int dinic()
{
int ans=0;
while (bfs())
ans+=dfs(s,inf);
return ans;
}
int main()
{
scanf("%d%d",&k,&n);
s=0;t=n+k+1;ans=0;
for (int i=1;i<=k;i++)
{
scanf("%d",&x);
ans+=x;
add(i+n,t,x);
}
for (int i=1;i<=n;i++)
{
scanf("%d",&m);
add(s,i,1);
while (m--)
{
scanf("%d",&x);
add(i,x+n,1);
}
}
int ant = dinic();
if (ant != ans)
printf("No Solution!");
else
{
for (int i=1;i<=n;i++)
{
for (int j=head[i];j>0;j=g[j].next)
{
if ((g[j].to!=s) && (g[j].val==0))
{
f[g[j].to-n][i]=1;
}
}
}
for (int i=1;i<=k;i++)
{
printf("%d:",i);
for (int j=1;j<=n;j++)
{
if (f[i][j]) printf(" %d",j);
}
printf("\n");
}
}
return 0;
}