POJ 3422 Kaka's Matrix Travels

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<iostream>
#include<queue>
using namespace std;
#define MAX 100
#define MAXN 100001
#define MAXE 1000001
#define IMAX 21474836
struct GROUPS{int f,t,flow,cost,next;};
GROUPS a[MAXE];
int N,K,map[MAX][MAX],tot=0,last[MAXN],dist[MAXN],ans=0,fa[MAXN];
bool vis[MAXN];
void add(int from,int to,int cost,int flow)
{
      a[tot].f=from;
      a[tot].t=to;
      a[tot].next=last[from];
      a[tot].flow=flow;
      a[tot].cost=cost;
      last[from]=tot++;
      
      a[tot].f=to;
      a[tot].t=from;
      a[tot].next=last[to];
      a[tot].flow=0;
      a[tot].cost=-cost;
      last[to]=tot++;      
}
void pre_group()
{
      for(int i=1;i<=N;i++)
            for(int j=1;j<=N;j++)
            {
                  int getnum=(i-1)*N+j;
                  add(getnum,N*N+getnum,-map[i][j],1);
                  add(getnum,N*N+getnum,0,K-1);
            }
      for(int i=1;i<N;i++)//down to conect
            for(int j=1;j<=N;j++)
            {
                  int nextnum=i*N+j;
                  int nownum=(i-1)*N+j;
                  add(N*N+nownum,nextnum,0,K);
            }
      for(int i=1;i<=N;i++)//right to conect
            for(int j=1;j<N;j++)
            {
                  int nextnum=(i-1)*N+j+1;
                  int nownum=(i-1)*N+j;
                  add(N*N+nownum,nextnum,0,K);
            }
      add(0,1,0,K);
      add(2*N*N,2*N*N+1,0,K);          
}
bool spfa()
{
      queue<int> Q;
      for(int i=0;i<=2*N*N+1;i++)
            dist[i]=IMAX;
      memset(fa,-1,sizeof(fa));
      memset(vis,false,sizeof(vis));
      dist[0]=0;
      vis[0]=true;
      Q.push(0);
      while(!Q.empty())
      {
            int now=Q.front();
            Q.pop();
            for(int i=last[now];i!=-1;i=a[i].next)
            {
                  int to=a[i].t;
                  if(dist[now]+a[i].cost<dist[to] && a[i].flow)
                  {
                        dist[to]=dist[now]+a[i].cost;
                        fa[to]=i;
                        if(!vis[to])
                        {
                              vis[to]=true;
                              Q.push(to);
                        }
                  }
            }
            vis[now]=false;
      }
      if(dist[2*N*N+1]==IMAX)   return false;
      return true;
}
void work()
{
      while(spfa())
      {
            int minflow=IMAX;
            for(int i=fa[2*N*N+1];i!=-1;i=fa[a[i].f])
                  minflow=minflow>a[i].flow?a[i].flow:minflow;
            for(int i=fa[2*N*N+1];i!=-1;i=fa[a[i].f])
            {
                  a[i].flow-=minflow;
                  a[i^1].flow+=minflow;
            }
            ans+=dist[2*N*N+1]*minflow;
      }
}
int main()
{
      //freopen("input.in","r",stdin);
	  //freopen("output.out","w",stdout); 
	  memset(last,-1,sizeof(last));
	  scanf("%d%d",&N,&K);
	  for(int i=1;i<=N;i++)
	        for(int j=1;j<=N;j++)
	              scanf("%d",&map[i][j]);
      pre_group();
      work();
      printf("%d",-ans);
	  //system("pause");
      return 0;
}

—————————————————————————————————

本文原创自Slience的csdn技术博客。

本博客所有原创文章请以链接形式注明出处。

欢迎关注本技术博客，本博客的文章会不定期更新。

大多数人想要改造这个世界，但却罕有人想改造自己。

世上没有绝望的处境，只有对处境绝望的人。

                                              ————By slience

—————————————————————————————————