HOJ - 2715最小费用流

国庆八天乐，刷题也快乐。

HOJ崩了，但是VJ可以把题目挂出来。

题目链接：https://vjudge.net/contest/188441#problem/A

涉及到矩阵里面的网络流，化为图来做。

某个点有流量限制，一定要想到拆点。

求最大值的话，要把w变成负数。

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <vector>
 #include <queue>
 #include <algorithm>
 using namespace std;
 const int maxn = 6e3;
 const int INF = 1e9;
 int vis[maxn],dist[maxn];
 int tot,head[maxn];
 int pv[maxn],pe[maxn];
 struct edge
 {
     int to,pre,cap,cost;
 }e[];
 void init()
 {
     tot = ;
     memset(head,-,sizeof(head));
 }
 void add(int from,int to,int cap,int cost)
 {
     e[tot].pre = head[from];
     e[tot].to = to;
     e[tot].cap = cap;
     e[tot].cost = cost;
     head[from] = tot++;
 }
 void addedge(int from,int to,int cap,int cost)
 {
     add(from,to,cap,cost);
     add(to,from,,-cost);
 }
 int n,k;
 int min_cost_flow(int s,int t,int f,int& max_flow)
 {
     int ret = ;
     while(f>&&(k--))
     {
         memset(vis,,sizeof(vis));
         for(int i=;i<maxn;i++) dist[i] = INF;
         dist[s] = ;
         queue<int> q;
         q.push(s);
         while(!q.empty())
         {
             int v = q.front(); q.pop();
             vis[v] = ;
             for(int i=head[v];i>=;i=e[i].pre)
             {
                 int to = e[i].to,cap = e[i].cap,cost = e[i].cost;
                 if(cap>&&dist[to]>dist[v]+cost)
                 {
                     pv[to] = v,pe[to] = i;
                     dist[to] = dist[v] + cost;
                     if(!vis[to]) q.push(to);
                     vis[to] = ;
                 }
             }
            // printf("%d\n",v);
         }
         //printf("%d\n",dist[t]);
         if(dist[t]==INF) return ret;///同一目的地，每次增广路都是最小费用
         ///当所有边的流量都流净后，即没有残余网络，返回。
         int d = f;
         for(int v=t;v!=s;v=pv[v])
         {
             d = min(d,e[pe[v]].cap);
         }
         f -= d;
         max_flow += d;
         ret += d*dist[t]; ///走一单位就消耗dist[t]
         for(int v=t;v!=s;v=pv[v])
         {
             e[pe[v]].cap -= d;
             e[pe[v]^].cap += d;
         }
     }
     return ret;
 }
 int height[][];
 int value[][];
 int judge(int x,int y)
 {
     if(x<=||x>=(n-)||y<=||y>=(n-)) return ;
     else return ;
 }
 int judge1(int x,int y)
 {
     if(x>=&&x<=(n-)&&y>=&&y<=(n-)) return ;
     else return ;
 }
 int to[][] = {,,-,,,,,-};
 int main()
 {
     int T;scanf("%d",&T);
     while(T--)
     {
         scanf("%d %d",&n,&k);
         init();  ///千万别忘记写
         for(int i=;i<n;i++)
         {
             for(int j=;j<n;j++)
             {
                 scanf("%d",&value[i][j]);
             }
         }
         for(int i=;i<n;i++)
         {
             for(int j=;j<n;j++)
             {
                 scanf("%d",&height[i][j]);
             }
         }
         int s = *n*n+;
         int t = *n*n+;
         for(int i=;i<n;i++)
         {
             for(int j=;j<n;j++)
             {
                 addedge(s,i*n+j,INF,);
                 addedge(i*n+j,i*n+j+n*n,,-value[i][j]);
                 addedge(i*n+j,i*n+j+n*n,INF,);  ///addedge(i',i'',inf,0)的边，可以再次走，只不过没有宝藏了
                 if(judge(i,j))
                 {
                     addedge(i*n+j+n*n,t,INF,);
                 }
             }
         }
         for(int i=;i<n;i++)
         {
             for(int j=;j<n;j++)
             {
                 for(int k=;k<;k++)
                 {
                     int nextx = i+to[k][];
                     int nexty = j+to[k][];
                     if(judge1(nextx,nexty)&&height[i][j]>height[nextx][nexty])
                     {
                         addedge(i*n+j+n*n,nextx*n+nexty,INF,);
                     }
                 }
             }
         }
         int max_flow = ;
         int ans = min_cost_flow(s,t,INF,max_flow);
         if(ans==)
         {
             printf("%d\n",ans);
         }
         else
         {
             printf("%d\n",-ans);
         }
     }
     return ;
 }
 /*
 4 5
 1 2 1
 2 3 1
 3 4 1
 1 3 2
 2 4 1
 */