bzoj 2127 happiness

2127: happiness
Time Limit: 51 Sec Memory Limit: 259 MB
Submit: 1889 Solved: 922
[Submit][Status][Discuss]
Description

高一一班的座位表是个n*m的矩阵，经过一个学期的相处，每个同学和前后左右相邻的同学互相成为了好朋友。这学期要分文理科了，每个同学对于选择文科与理科有着自己的喜悦值，而一对好朋友如果能同时选文科或者理科，那么他们又将收获一些喜悦值。作为计算机竞赛教练的scp大老板，想知道如何分配可以使得全班的喜悦值总和最大。
Input

第一行两个正整数n，m。接下来是六个矩阵第一个矩阵为n行m列 此矩阵的第i行第j列的数字表示座位在第i行第j列的同学选择文科获得的喜悦值。第二个矩阵为n行m列 此矩阵的第i行第j列的数字表示座位在第i行第j列的同学选择理科获得的喜悦值。第三个矩阵为n-1行m列 此矩阵的第i行第j列的数字表示座位在第i行第j列的同学与第i+1行第j列的同学同时选择文科获得的额外喜悦值。第四个矩阵为n-1行m列 此矩阵的第i行第j列的数字表示座位在第i行第j列的同学与第i+1行第j列的同学同时选择理科获得的额外喜悦值。第五个矩阵为n行m-1列 此矩阵的第i行第j列的数字表示座位在第i行第j列的同学与第i行第j+1列的同学同时选择文科获得的额外喜悦值。第六个矩阵为n行m-1列 此矩阵的第i行第j列的数字表示座位在第i行第j列的同学与第i行第j+1列的同学同时选择理科获得的额外喜悦值。
Output

输出一个整数，表示喜悦值总和的最大值
Sample Input
1 2

1 1

100 110

1

1000

Sample Output
1210

【样例说明】

两人都选理，则获得100+110+1000的喜悦值。

【数据规模】

对于100%以内的数据，n,m<=100 所有喜悦值均为小于等于5000的非负整数

---

【分析】
网络流最小割神奇建图法第二弹
题目要求最大，可以把给定的喜悦值取相反数，使问题变为总和最小。
每个同学自身对文理科的喜悦值最后加上，先考虑二元关系：
一对好朋友同时选择文科，获得 -v1的喜悦值。
一对好朋友同时选择理科，获得 –v2的喜悦值。
尝试最小割模型，割开与S相连的边表示选文科，割开与T相连的边表示选理科。
解方程：e=K/2=(v1+v2)/2.
令a= -v1/2，解出b= -v1/2, c= -v2/2, d= -v2/2.
a,b,c,d<0，令它们同时加上(v1+v2)/2，得到a=b=v2/2, c=d=v1/2.
把每个同学本身选择文、理科的喜悦值分别加入它与S、T之间的边。
为了避免分数，令所有边的容量乘2，求最小割。
求出的结果除以2，再减去M*(v1+v2)，M为二元关系个数。

以上是lyd式做法。

还有不这么麻烦的…就是把思路再逆一下，先把所有题目给出的权值加起来，累加进变量ans中，再用lyd式建图法，把负的都改成正的，然后求最小割tmp，注意乘2和除以2，然后用ans-tmp/2(先乘2再除以2是要防止小数出现)

---

【代码】

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#define fo(i,j,k) for(i=j;i<=k;i++)
#define p(i,j) (i-1)*100+j
using namespace std;
const int mxn=100005;
queue <int> q;
int n,m,s,t,cnt,ans,tmp,tans;
int head[mxn],a[105][105],b[105][105],dis[mxn];
struct node {int to,flow,next;} f[mxn<<4];
inline void add(int u,int v,int flow)
{
    f[++cnt].to=v,f[cnt].next=head[u],f[cnt].flow=flow,head[u]=cnt;
    f[++cnt].to=u,f[cnt].next=head[v],f[cnt].flow=0,head[v]=cnt;
}
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
inline bool bfs()
{
    memset(dis,-1,sizeof dis);
    q.push(s);
    dis[s]=0;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=head[u];i;i=f[i].next)
        {
            int v=f[i].to;
            if(dis[v]<0 && f[i].flow>0)
            {
                dis[v]=dis[u]+1;
                q.push(v);
            }
        }
    }
    if(dis[t]>0) return 1;
    return 0;
}
inline int find(int u,int now)
{
    int a,v,i,sum=0;
    if(u==t) return now;
    for(i=head[u];i;i=f[i].next)
    {
        v=f[i].to;
        if(f[i].flow>0 && now>sum && dis[v]==dis[u]+1 && (a=find(v,min(now-sum,f[i].flow))))
        {
            f[i].flow-=a;
            if(i&1) f[i+1].flow+=a;
            else f[i-1].flow+=a;
            sum+=a;
        }
    }
    if (!sum) dis[u]=-1;
    return sum;
}
int main()
{
    int i,j,u,v,w;
    n=read(),m=read();
    t=10*n*m;
    fo(i,1,n) fo(j,1,m) w=read(),add(s,p(i,j),w<<1),ans+=w;
    fo(i,1,n) fo(j,1,m) w=read(),add(p(i,j),t,w<<1),ans+=w;
    fo(i,1,n-1) fo(j,1,m) a[i][j]=read();
    fo(i,1,n-1) fo(j,1,m) b[i][j]=read();
    fo(i,1,n-1) fo(j,1,m)
    {
        add(s,p(i,j),a[i][j]),add(s,p(i+1,j),a[i][j]);
        add(p(i,j),t,b[i][j]),add(p(i+1,j),t,b[i][j]);
        add(p(i,j),p(i+1,j),a[i][j]+b[i][j]),add(p(i+1,j),p(i,j),a[i][j]+b[i][j]);
        ans+=a[i][j]+b[i][j];
    }
    fo(i,1,n) fo(j,1,m-1) a[i][j]=read();
    fo(i,1,n) fo(j,1,m-1) b[i][j]=read();
    fo(i,1,n) fo(j,1,m-1)
    {
        add(s,p(i,j),a[i][j]),add(s,p(i,j+1),a[i][j]);
        add(p(i,j),t,b[i][j]),add(p(i,j+1),t,b[i][j]);
        add(p(i,j),p(i,j+1),a[i][j]+b[i][j]),add(p(i,j+1),p(i,j),a[i][j]+b[i][j]);
        ans+=a[i][j]+b[i][j];
    }
    while(bfs()) tmp+=find(s,0x7f7f7f7f);
    printf("%d\n",ans-(tmp>>1));
    return 0;
}