HDU1569 方格取数(2) 网络流

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<algorithm>
#include<memory.h>
#include<queue>
#include<stack>

using namespace std;

#define inf 0x3fffffff
int n,m,cnt,s,t;
int map[51][51],head[3000],dis[3000],gap[3000];

struct EDGE
{
    int to,from,next;
}edge[15000];

int min(int a,int b)
{
    return a<b?a:b;
}

void init()
{
    cnt=0;
    memset(head,-1,sizeof(head));
    memset(gap,0,sizeof(gap));
    memset(dis,0,sizeof(dis));
    gap[0]=t;
}

void add(int x,int y,int c)
{
    edge[cnt].to=y;
    edge[cnt].from=c;
    edge[cnt].next=head[x];
    head[x]=cnt++;
}


int dfs(int x,int flow)
{
    if(x==t)
		return flow;
    int temp=flow;
    int pos=t-1;
    int j;
    for(j=head[x];j!=-1;j=edge[j].next)
    {
        int y=edge[j].to;
        int c=edge[j].from;
        if(c>0 && dis[x]==dis[y]+1)
        {
            int temp_flow=dfs(y,min(temp,c));
            temp-=temp_flow;
            edge[j].from-=temp_flow;
            edge[j^1].from+=temp_flow;
            if(temp==0 || dis[s]==t)
				return flow-temp;
        }
        if(c>0 && pos>dis[y])
			pos=dis[y];
    }
    if(temp==flow)
    {
        if(!(--gap[dis[x]]))
            dis[s]=t;
        else
            gap[dis[x]=pos+1]++;
    }
    return flow-temp;
}

int sap()
{
    int maxflow=0;
    while(dis[s]<t)
    {
        maxflow+=dfs(s,inf);
    }
    return maxflow;
}

int main()
{
	int i,j;
    while(cin>>n>>m)
    {
        s=n*m+1,t=s+1;
        init();
        int sum=0;
        for(i=1;i<=n;i++)
			for( j=1;j<=m;j++)
			{
				scanf("%d",&map[i][j]);
				sum+=map[i][j];
			}
			for( i=1;i<=n;i++)
				for(int j=1;j<=m;j++)
				{
					int num=(i-1)*m+j;
					if((i+j)%2)
					{
						add(s,num,map[i][j]);
						add(num,s,0);
						if(i>1)
						{
							add(num,num-m,inf);
							add(num-m,num,0);
						}
						if(j>1)
						{
							add(num,num-1,inf);
							add(num-1,num,0);
						}
						if(i<n)
						{
							add(num,num+m,inf);
							add(num+m,num,0);
						}
						if(j<m)
						{
							add(num,num+1,inf);
							add(num+1,num,0);
						}
						
					}
					else
					{
						add(num,t,map[i][j]);
						add(t,num,0);
					}
				}
				printf("%d\n",sum-sap());
    }
	return 0;
}