HDU 2686 && HDU 3376（网络流之费用流）

题目地址：HDU 2686       HDU 3376

这两道题目除了数据大小外是一样的。前者只有30*30，但是后者却成了600*600。。本来以为前者代码用到后者会超时，迟迟没敢交，但是感觉能用费用流的话也只能这么做了，于是改了改数组大小就交上去了。还真没超时。。

这题又是一道关于来回最短路的。最大费用可以把费用改成相反数，最后再转成相反数就是最大费用了。

建图思路是拆点，限制每个点只能经过一次。然后将每个点与右边的和下边的连边。源点与汇点要设为2个流量。

不知道为什么用G++叫就一直WA，换成C++就AC了。。。

代码如下；

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include<algorithm>

using namespace std;
const int INF=0x3f3f3f3f;
int head[800000], source, sink, cost, flow, cnt, mp[700][700];
int cur[800000], d[800000], pre[800000], vis[800000];
struct node
{
int u, v, cap, next, cost;
}edge[10000000];
void add(int u, int v ,int cap, int cost)
{
edge[cnt].v=v;
edge[cnt].cap=cap;
edge[cnt].cost=cost;
edge[cnt].next=head[u];
head[u]=cnt++;

edge[cnt].v=u;
edge[cnt].cap=0;
edge[cnt].cost=-cost;
edge[cnt].next=head[v];
head[v]=cnt++;
}
int spfa()
{
memset(d,INF,sizeof(d));
memset(vis,0,sizeof(vis));
queue<int>q;
q.push(source);
cur[source]=-1;
d[source]=0;
int minflow=INF, i;
while(!q.empty())
{
int u=q.front();
q.pop();
vis[u]=0;
for(i=head[u];i!=-1;i=edge[i].next)
{
int v=edge[i].v;
if(d[v]>d[u]+edge[i].cost&&edge[i].cap)
{
d[v]=d[u]+edge[i].cost;
minflow=min(minflow,edge[i].cap);
cur[v]=i;
if(!vis[v])
{
vis[v]=1;
q.push(v);
}
}
}
}
if(d[sink]==INF) return 0;
flow+=minflow;
cost-=minflow*d[sink];
for(i=cur[sink];i!=-1;i=cur[edge[i^1].v])
{
edge[i].cap-=minflow;
edge[i^1].cap+=minflow;
}
}
void mcmf(int n)
{
cost=0;
flow=0;
while(spfa()) ;
printf("%d\n",cost-mp[0][0]-mp[n-1][n-1]);
}
int main()
{
int n, i, j, k;
while(scanf("%d",&n)!=EOF)
{
memset(head,-1,sizeof(head));
cnt=0;
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
{
scanf("%d",&mp[i][j]);
}
}
source=2*n*n;
sink=2*n*n+1;
add(source,0,2,0);
add(2*n*n-1,sink,2,0);
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
{
if((i==0&&j==0)||(i==n-1&&j==n-1))
{
add(i*n+j,i*n+j+n*n,2,-mp[i][j]);
}
else
add(i*n+j,i*n+j+n*n,1,-mp[i][j]);
if(i<n-1)
add(i*n+j+n*n,(i+1)*n+j,1,0);
if(j<n-1)
add(i*n+j+n*n,i*n+j+1,1,0);
}
}
mcmf(n);
}
return 0;
}