【NOIP2013模拟联考2】摘取作物(pick)

Description

Feather的农场里有N*M块地，排列成N行，每行M块地。Feather在每块地里种植了不同的农作物。现在这些农作物都成熟了，可以摘取下来出售了。其中第i行第j列的地里的农作物的价值为W[i,j]。

JackRabbit是Feather的好友，平时经常为Feather的农作物除草除虫。为了答谢JackRabbit，Feather决定把一部分农作物送给JackRabbit。JackRabbit很高兴，恨不得一下子把农场里的农作物摘空。

为了防止JackRabbit把农作物摘空，Feather提出了两个条件：

1.每行最多选取两块地；

2.每列最多选取两块地。

这下子把JackRabbit难住了。如何在满足这两个条件的前提下，使得摘取的农作物的价值之和最大呢？

Analysis

首先分析题目

The essence

这是一道典型的费用流题

The solution

构造

从原点s出发，到左边N个点的容量为2，费用为0，表示每行最多选2块地。
中间有N*M条边连到右边M个点，容量为1，费用为a[i,j]，
表示每块地只能选1次，作物价值为a[i,j]。
右边M个点连到汇点t，容量为2，费用为0，表示每列最多选2块地

处理

每次找最大费用增广路，直到没有费用大于0的增广路为止。
此时的费用之和即为作物价值之和的最大值。

至于具体的细节处理，可以用spfa判断这可不可流通，可行的话就直接跑最大费用流了。这也是典型处理。

CODE

#include <cstdio>
#include <iostream>
#include <cmath>
#include <algorithm>
#include <cstring>
#define fo(i,a,b) for (int i=a;i<=b;i++)
#define fd(i,a,b) for (int i=a;i>=b;i--)
#define N 2005
#define INF 2147483647
using namespace std;
bool bz[N];
int t[N],next[N],last[N],f[N],dis[N],d[N],a[N],c[N];
int n,m,l=1,tot=0,ans=0;
void add(int x,int y,int z,int k)
{
    t[++l]=y; 
    next[l]=last[x]; 
    last[x]=l; 
    f[l]=z; 
    a[l]=k; 
}
bool spfa()
{
    int l=0,r=1;
    memset(bz,false,sizeof(bz));
    memset(dis,128,sizeof(dis));
    dis[0]=0;d[1]=0;bz[0]=true;
    while (l<r)
    {
        int now=d[++l];
        for (int k=last[now];k;k=next[k])
            if (dis[t[k]]<dis[now]+a[k] && f[k]>0)
            {
                dis[t[k]]=dis[now]+a[k];
                c[t[k]]=k;
                if (!bz[t[k]])
                {
                    bz[t[k]]=true;
                    d[++r]=t[k];
                }
            }
        bz[now]=false;
    }
    if (dis[tot]>0) ans+=dis[tot];
    return (dis[tot]>0);
}
void find()
{
    int x=tot,sum=INF;
    while (x!=0) sum=min(sum,f[c[x]]),x=t[c[x]^1]; x=tot;
    while (x!=0) f[c[x]]-=sum,f[c[x]^1]+=sum,x=t[c[x]^1];
}
int main()
{
    freopen("pick.in","r",stdin);
    freopen("pick.out","w",stdout);
    scanf("%d%d",&n,&m);
    tot=n*m+1;
    fo(i,1,n) add(0,i,2,0),add(i,0,0,0);
    fo(i,1,m) add(i+n,tot,2,0),add(tot,i+n,0,0);
    fo(i,1,n)
        fo(j,1,m)
        {
            int x;
            scanf("%d",&x);
            add(i,j+n,1,x);
            add(j+n,i,0,-x);
        }
    while (spfa()) find();
    printf("%d",ans);   
    return 0;
}