【ACM-ICPC 2018 南京现场赛 】 I.Magic Potion（网络最大流）

题意

有 n 个英雄，m 个怪兽， k 瓶药水，接下来n行代表第 i 个人可以杀的怪兽编号，每个人只能击败一只怪兽，但是如果有药水就可以杀两只怪兽，但是每个人只能用一次药水。 问最多能杀多少只怪兽。

分析

可以用网络最大流做，难点就在建图时如何连点。

解法一

源点连每个英雄，流通量为1；英雄连怪兽，流通量为1；怪兽连汇点，流通量为1；源点再连一个中转点出来，流通量为 k ；中转点连每个英雄，流通量为1。只需要跑一遍dinic。

样例1：

仔细思考一下，挺巧妙的。

解法二

源点连每个英雄，流通量为1；英雄连怪兽，流通量为1；怪兽连汇点，流通量为1。先跑一遍 dinic，得到的是都不用药水时的最大流 a ，然后假设我们给每一个英雄一瓶药水（即再给源点向英雄 1 的流量），这时候再跑一遍 dinic，得到每个英雄再最多杀一个怪兽能击败的怪兽数量 b，如果 b < k，那么答案就是 a + b 因为最多就只能再击败 b 个怪物；相反，如果 b >= k，答案就是 a + k，因为最多只有 k 瓶药水。

样例1：

---

解法一代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#define inf 0x3f3f3f3f3f3f3f3f
#define ll long long
#define MAXN 10005
using namespace std;
int n,m,k;
int sp,tp;
struct node
{
    int v,next;
    ll cap;
} mp[MAXN*10];
int pre[MAXN],dis[MAXN],cur[MAXN];
int cnt=0;
void init()
{
    cnt=0;
    memset(pre,-1,sizeof(pre));
}
void add(int u,int v,int w)
{
    mp[cnt].v=v;
    mp[cnt].cap=w;
    mp[cnt].next=pre[u];
    pre[u]=cnt++;
}
bool bfs()
{
    memset(dis,-1,sizeof(dis));
    queue<int>q;
    while(!q.empty())
        q.pop();
    q.push(sp);
    dis[sp]=0;
    int u,v;
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        for(int i=pre[u]; i!=-1; i=mp[i].next)
        {
            v=mp[i].v;
            if(dis[v]==-1&&mp[i].cap>0)
            {
                dis[v]=dis[u]+1;
                q.push(v);
                if(v==tp)
                    break;
            }
        }
    }
    return dis[tp]!=-1;
}
ll dfs(int u,ll cap)
{
    if(u==tp||cap==0)
        return cap;
    ll res=0,f;
    for(int &i=cur[u]; i!=-1; i=mp[i].next)
    {
        int v=mp[i].v;
        if(dis[v]==dis[u]+1&&(f=dfs(v,min(cap-res,mp[i].cap)))>0)
        {
            mp[i].cap-=f;
            mp[i^1].cap+=f;
            res+=f;
            if(res==cap)
                return cap;
        }
    }
    if(!res)
        dis[u]=-1;
    return res;
}
ll dinic()
{
    ll ans=0;
    while(bfs())
    {
        for(int i=0; i<=n+m+2; i++)
            cur[i]=pre[i];
        ans+=dfs(sp,inf);
    }
    return ans;
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        int u,v,t;
        init();
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&t);
            for(int j=1;j<=t;j++) {
                scanf("%d",&u);
                add(i,n+u,1);
                add(n+u,i,0);
            }
        }
        for(int i=1;i<=m;i++) {
            add(n+i,n+m+2,1);
            add(n+m+2,n+i,0);
        }
        for(int i=1;i<=n;i++) {
            add(0,i,1);
            add(i,0,0);
        }
        add(0,n+m+1,k);//建立中转点 
        add(n+m+1,0,0);
        for(int i=1;i<=n;i++) {
            add(n+m+1,i,1);
            add(i,n+m+1,0);
        }
        sp=0;
        tp=n+m+2;
        ll ans=dinic();
        
        printf("%lld\n",ans);
    }
    return 0;
}

解法二代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#define inf 0x3f3f3f3f3f3f3f3f
#define ll long long
#define MAXN 10005
using namespace std;
int n,m,k;
int sp,tp;
struct node
{
    int v,next;
    ll cap;
} mp[MAXN*10];
int pre[MAXN],dis[MAXN],cur[MAXN];
int cnt=0;
void init()
{
    cnt=0;
    memset(pre,-1,sizeof(pre));
}
void add(int u,int v,int w)
{
    mp[cnt].v=v;
    mp[cnt].cap=w;
    mp[cnt].next=pre[u];
    pre[u]=cnt++;
}
bool bfs()
{
    memset(dis,-1,sizeof(dis));
    queue<int>q;
    while(!q.empty())
        q.pop();
    q.push(sp);
    dis[sp]=0;
    int u,v;
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        for(int i=pre[u]; i!=-1; i=mp[i].next)
        {
            v=mp[i].v;
            if(dis[v]==-1&&mp[i].cap>0)
            {
                dis[v]=dis[u]+1;
                q.push(v);
                if(v==tp)
                    break;
            }
        }
    }
    return dis[tp]!=-1;
}
ll dfs(int u,ll cap)
{
    if(u==tp||cap==0)
        return cap;
    ll res=0,f;
    for(int &i=cur[u]; i!=-1; i=mp[i].next)
    {
        int v=mp[i].v;
        if(dis[v]==dis[u]+1&&(f=dfs(v,min(cap-res,mp[i].cap)))>0)
        {
            mp[i].cap-=f;
            mp[i^1].cap+=f;
            res+=f;
            if(res==cap)
                return cap;
        }
    }
    if(!res)
        dis[u]=-1;
    return res;
}
ll dinic()
{
    ll ans=0;
    while(bfs())
    {
        for(int i=0; i<=n+m+2; i++)
            cur[i]=pre[i];
        ans+=dfs(sp,inf);
    }
    return ans;
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        int u,v,t;
        init();
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&t);
            for(int j=1;j<=t;j++) {
                scanf("%d",&u);
                add(i,n+u,1);
                add(n+u,i,0);
            }
        }
        for(int i=1;i<=m;i++) {
            add(n+i,n+m+2,1);
            add(n+m+2,n+i,0);
        }
        for(int i=1;i<=n;i++) {
            add(0,i,1);
            add(i,0,0);
        }
        //add(0,n+m+1,k);
        //add(n+m+1,0,0);
        for(int i=1;i<=n;i++) {
            add(n+m+1,i,1);
            add(i,n+m+1,0);
        }
        sp=0;
        tp=n+m+2;
        ll ans=dinic();
        
        for(int i=pre[sp];i!=-1;i=mp[i].next)//重建源点向英雄的边 
        {
        	mp[i].cap = 1;
		}
		
		ll used=dinic();
		if(used < k) ans += used;
		else ans += k;
        printf("%lld\n",ans);
    }
    return 0;
}