bzoj 3894 文理分科

3894: 文理分科

Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 869 Solved: 512
[Submit][Status][Discuss]
Description

文理分科是一件很纠结的事情！（虽然看到这个题目的人肯定都没有纠
结过）
小P所在的班级要进行文理分科。他的班级可以用一个n*m的矩阵进行
描述，每个格子代表一个同学的座位。每位同学必须从文科和理科中选择
一科。同学们在选择科目的时候会获得一个满意值。满意值按如下的方式
得到：
1．如果第i行第秒J的同学选择了文科，则他将获得art[i][j]的满意值，如
果选择理科，将得到science[i][j]的满意值。
2．如果第i行第J列的同学选择了文科，并且他相邻（两个格子相邻当且
仅当它们拥有一条相同的边）的同学全部选择了文科，则他会更开
心，所以会增加same_art[i][j]的满意值。
3．如果第i行第j列的同学选择了理科，并且他相邻的同学全部选择了理
科，则增加same_science[i]j[]的满意值。
小P想知道，大家应该如何选择，才能使所有人的满意值之和最大。请
告诉他这个最大值。
Input

第一行为两个正整数：n，m
接下来n术m个整数，表示art[i][j]；
接下来n术m个整数．表示science[i][j]；
接下来n术m个整数，表示same_art[i][j]；
Output

输出为一个整数，表示最大的满意值之和
Sample Input

3 4

13 2 4 13

7 13 8 12

18 17 0 5

8 13 15 4

11 3 8 11

11 18 6 5

1 2 3 4

4 2 3 2

3 1 0 4

3 2 3 2

0 2 2 1

0 2 4 4
Sample Output

152
HINT

样例说明

1表示选择文科，0表示选择理科，方案如下：

1 0 0 1

0 1 0 0

1 0 0 0

N,M<=100,读入数据均<=500

---

【分析】
题目大意：给定一个m*n的矩阵，每个格子的人可以学文或者学理，学文和学理各有一个满意度，如果以某人为中心的十字内所有人都学文或者学理还会得到一个额外满意度，求最大满意度之和
令S集为学文，T集为学理
每个人学文或者学理的满意度很好连边
如果某个集合内的人都学理会获得一个满意度，那么就新加一个点，将集合内的所有人向这个点连流量为正无穷的边，再从这个点向T连一条流量为满意度的边，表示集合内任意一个人学文都要把这个点与T的边割掉
都学文同理
建完图之后跑最小割即可

天上地下 只服尹神

---

【代码】

/**************************************************************
    Problem: 3894
    User: chaijing
    Language: C++
    Result: Accepted
    Time:5372 ms
    Memory:7548 kb
****************************************************************/

//bzoj 3894 文理分科 
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#define inf 1e9+7
#define fo(i,j,k) for(i=j;i<=k;i++)
#define p(i,j) (i-1)*m+j
using namespace std;
queue <int> q;
const int mxn=100005;
int n,m,cnt,ans,s,t,tot;
int head[mxn],dis[mxn],math[mxn],art[mxn];
struct node {int to,next,flow;} f[mxn<<2]; 
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 bool bfs()
{
    int i,j,u,v,flow;
    memset(dis,-1,sizeof dis);
    q.push(s);
    dis[s]=0;
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        for(i=head[u];i;i=f[i].next)
        {
            v=f[i].to,flow=f[i].flow;
            if(dis[v]==-1 && flow>0)
              dis[v]=dis[u]+1,q.push(v);
        }
    }
    if(dis[t]>0) return 1;
    return 0;
}
inline int find(int u,int low)
{
    int v,i,j,a=0,sum=0,flow;
    if(u==t) return low;
    for(i=head[u];i;i=f[i].next)
    {
        v=f[i].to,flow=f[i].flow;
        if(flow>0&& dis[v]==dis[u]+1 && (a=find(v,min(low-sum,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,k1,k2,k3,k4,u,v,x,flow;
    scanf("%d%d",&n,&m);
    s=0,t=m*n+1,tot=t;
    fo(i,1,n) fo(j,1,m)   //文 
    {
        int tmp=p(i,j);
        scanf("%d",&art[tmp]);
        ans+=art[tmp];
        add(s,tmp,art[tmp]);
    }
    fo(i,1,n) fo(j,1,m)   //理 
    {
        int tmp=p(i,j);
        scanf("%d",&math[tmp]);
        ans+=math[tmp];
        add(tmp,t,math[tmp]);
    }
    fo(i,1,n) fo(j,1,m)   //同文 
    {
        tot++;
        int tmp=p(i,j);
        if(i>1) k1=p(i-1,j);else k1=0;
        if(i<n) k2=p(i+1,j);else k2=0;
        if(j>1) k3=p(i,j-1);else k3=0;
        if(j<m) k4=p(i,j+1);else k4=0;
        scanf("%d",&x);ans+=x;
        add(tot,tmp,inf);
        if(k1) add(tot,k1,inf);
        if(k2) add(tot,k2,inf);
        if(k3) add(tot,k3,inf);
        if(k4) add(tot,k4,inf);
        add(s,tot,x);
    }
    fo(i,1,n) fo(j,1,m)   //同理  
    {
        tot++;
        int tmp=p(i,j);
        if(i>1) k1=p(i-1,j);else k1=0;
        if(i<n) k2=p(i+1,j);else k2=0;
        if(j>1) k3=p(i,j-1);else k3=0;
        if(j<m) k4=p(i,j+1);else k4=0;
        scanf("%d",&x);ans+=x;
        add(tmp,tot,inf);
        if(k1) add(k1,tot,inf);
        if(k2) add(k2,tot,inf);
        if(k3) add(k3,tot,inf);
        if(k4) add(k4,tot,inf);
        add(tot,t,x);
    }
    while(bfs())
      ans-=find(s,inf);
    printf("%d\n",ans);
    return 0;
}
//3 4
//13 2 4 13
//7 13 8 12
//18 17 0 5
//8 13 15 4
//11 3 8 11
//11 18 6 5
//1 2 3 4
//4 2 3 2
//3 1 0 4
//3 2 3 2
//0 2 2 1
//0 2 4 4