题意:n个英雄打m个怪兽,每个英雄只能打自己集合的怪兽,并且只能打一只,还有k个药水,一个英雄用一个药水能多打一只怪兽,问所有一共最多能打多少只怪兽
思路:一开始想的二分匹配,但药水实在不好分配,赛后看题解发现就是一简单网络流,还是题目做少了。
建图:1.源点和每个英雄连边,权值1,
2.源点和药水连边,权值为k
3.药水和每个英雄连边,权值为1
4.每个英雄和对应的怪兽连边,权值为1
5.每个怪兽和汇点连边,权值为1
然后最大流就是答案了
#include<stdio.h>
#include<queue>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
#define inf 1e9
#define maxn 1005
#define maxm 105000
struct Edge{
int to,next,flow;
} e[maxm];
int d1[maxn],n,m,head[maxn],cur[maxn],cnt,st,ed,k;
void add(int u,int v,int flow)
{
e[cnt] = (Edge)
{
v, head[u], flow
};
head[u] = cnt++;
e[cnt] = (Edge)
{
u, head[v], 0
};
head[v] = cnt++;
}
void init()
{
cnt=0;
st=0;
ed=n+m+2;
memset(head,-1,sizeof(head));
}
int dfs(int now,int ed,int lim)
{
if(now==ed||!lim)
{
return lim;
}
int flow=0,f1;
for(int i=cur[now]; i!=-1; i=e[i].next)
{
cur[now]=i;
int v=e[i].to;
if(d1[v]==d1[now]+1&&(f1=dfs(v,ed,min(e[i].flow,lim))))
{
flow+=f1;
lim-=f1;
e[i].flow-=f1;
e[i^1].flow+=f1;
if(!lim)
break;
}
}
return flow;
}
int bfs(int st,int ed)
{
for(int i=0; i<=ed; i++)
{
cur[i]=head[i];
d1[i]=0x7f7f7f7f;
}
queue<int>q;
d1[st]=0;
q.push(st);
while(!q.empty())
{
int u=q.front();
q.pop();
//printf("u=%d\n",u);
for(int i=head[u]; i!=-1; i=e[i].next)
{
int v=e[i].to;
//printf(" d[%d]=%d e[i].flow=%d\n",v,d[v],e[i].flow);
if(d1[v]>inf&&e[i].flow)
{
d1[v]=d1[u]+1;
q.push(v);
}
}
}
if(d1[ed]<inf)
return 1;
else
return 0;
}
int slove()
{
int ans1=0;
while(bfs(st,ed))
{
ans1+=dfs(st,ed,inf);
}
return ans1;
}
void buildmap()
{
init();
for(int i=1; i<=n; i++)
{
add(st,i,1);
int k;
scanf("%d",&k);
for(int j=1; j<=k; j++)
{
int x;
scanf("%d",&x);
add(i,x+n,1);
}
}
for(int i=1; i<=m; i++)
{
add(n+i,ed,1);
}
add(st,n+m+1,k);
for(int i=1;i<=n;i++)
{
add(n+m+1,i,1);
}
}
int main()
{
while(scanf("%d %d %d",&n,&m,&k)!=EOF)
{
buildmap();
int ans=slove();
printf("%d\n",ans);
}
}
/*
3 5 2
4 1 2 3 5
2 2 5
2 1 2
*/