HDU 1569——方格取数(2) 【不用vector】

小白书的用vector的好像在某些情况下特别慢。写了邻接表的Dinic当做模板。

#include<cstdio>
#include<cstring>
#include<queue>
#include<iostream>
using namespace std;
#define LL long long 
#define INF 100000000
#define maxn 2600
#define maxm 30000

struct Edge
{
	int u,v,cap,flow,next;
	Edge(){}
	Edge(int uu,int vv,int cc,int ff,int nn)
	{
		u=uu;v=vv;
		cap=cc;
		flow=ff;
		next=nn;
	}
}edge[maxm];
int head[maxn],cnt;
int n,m;
int s,t;
int map[maxn][maxn];
int dir[4][2]={1,0,-1,0,0,1,0,-1};
/*--bfs--*/
int vis[maxn],dis[maxn];
/*--dfs--*/
int cur[maxn];
/*-----------------------*/
int judge(int x,int y)
{
	if(x>=0&&x<n&&y>=0&&y<m)
		return 1;
	return 0;
}

void add(int u,int v,int cap)
{
	edge[cnt]=Edge(u,v,cap,0,head[u]);
	head[u]=cnt++;
	edge[cnt]=Edge(v,u,0,0,head[v]);
	head[v]=cnt++;
}
int bfs()
{
	memset(vis,0,sizeof(vis));
	queue<int> q;
	q.push(s);
	vis[s]=1;
	dis[s]=0;
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		for(int i=head[u];~i;i=edge[i].next)
		{
			Edge& e=edge[i];
			if(vis[e.v]==0&&e.cap>e.flow)
			{
				vis[e.v]=1;
				dis[e.v]=dis[u]+1;
				q.push(e.v);
			}
		}
	}
	return vis[t];
}
int dfs(int u,int low)
{
	if(u==t||low==0)
		return low;
	int flow=0,d;
	for(int &i=cur[u];~i;i=edge[i].next)
	{
		Edge& e=edge[i];
		if(dis[u]+1==dis[e.v]&&(d=dfs(e.v,min(low,e.cap-e.flow)))>0)
		{
			e.flow+=d;
			edge[i^1].flow-=d;
			flow+=d;
			low-=d;
			if(low==0)
				break;
		}
	}
	return flow;
}
int Dinic()
{
	int flow=0;
	while(bfs())
	{
		for(int i=0;i<=t;i++)
			cur[i]=head[i];
		flow+=dfs(s,INF);
	}
	return flow;
}
int main()
{
	while(cin>>n>>m)
	{
		s=n*m;
		t=n*m+1;
		int sum=0;
		memset(head,-1,sizeof(head));
		cnt=0;
		for(int i=0;i<n;i++)
			for(int j=0;j<m;j++)
			{
				scanf("%d",&map[i][j]);
				sum+=map[i][j];
			}
		for(int i=0;i<n;i++)
			for(int j=0;j<m;j++)
				if((i+j)&1)
				{
					add(s,i*m+j,map[i][j]);
					for(int d=0;d<4;d++)
					{
						int x=i+dir[d][0];
						int y=j+dir[d][1];
						if(judge(x,y))
							add(i*m+j,x*m+y,INF);
					}
				}
				else
					add(i*m+j,t,map[i][j]);
		cout<<sum-Dinic()<<endl;
	}
	return 0;
}