Codevs1033蚯蚓的游戏问题-费用流

将田地上的食物取负，求图的最小费用流即为最大费用流的相反数。（ps第一次听说还可以这样用
思路：用spfa求最短路，沿着最短路增广。

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cstdio>
#include<vector>
#include<queue>
#define maxn 0x3f3f3f3f
using namespace std;
struct edge{int to,cap,cost,rev;};
vector <edge> E[5000];
int ans,n,k,m,endd,start;
int way_p[5000],way_n[5000];
int num[100][100];

void add_edge(int from,int to,int cap,int cost){
     E[from].push_back((edge){to,cap,cost,E[to].size()});
     E[to].push_back((edge){from,0,-cost,E[from].size()-1});
}
void build_E()
{
     int sum=0,s=(2*m+n-1)*n/2;
     for(int i=0;i<n;i++){
             for(int j=0;j<i+m;j++){
                     sum++;
                     if(i==0){add_edge(0,sum,1,0);}
                     add_edge(sum,s+sum,1,-num[i][j]);
             }
     }
     endd=s+sum+1;
     sum=0;
     for(int i=0;i<n-1;i++){
             for(int j=0;j<i+m;j++){
                     sum++;
                     add_edge(sum+s,sum+i+m,1,0);
                     add_edge(sum+s,sum+i+m+1,1,0);
             }
     }
     for(int j=0;j<n+m-1;j++){
             sum++;
             add_edge(sum+s,endd,1,0);
     }
     start=endd+1;
     add_edge(start,0,k,0);
}

bool spfa()
{
     bool used[5000]={0};
     int dist[5000];
     for(int i=0;i<=start;i++) dist[i]=maxn;
     queue <int> que;
     que.push(start);
     used[start]=true;
     dist[start]=0;
     while(!que.empty()){
                         int now=que.front();que.pop();
                         for(int i=0;i<E[now].size();i++){
                                 edge &e=E[now][i];
                                 if(e.cap>0 && dist[e.to]>e.cost+dist[now]){
                                            dist[e.to]=e.cost+dist[now];
                                            way_p[e.to]=now;
                                            way_n[e.to]=i;
                                            if(used[e.to]==false){
                                                                  que.push(e.to);
                                                                  used[e.to]=true;
                                            }
                                 }
                         }
                         used[now]=false;
     }
     if(dist[endd]>=maxn) return false;
     return true;
}

void work()
{
      int now=endd,f=maxn;
      while(now!=start){
                        edge &e=E[way_p[now]][way_n[now]];
                        f=min(f,e.cap);
                        now=way_p[now];
      }
      now=endd;
      while(now!=start){
                        edge &e=E[way_p[now]][way_n[now]];
                        e.cap-=f;
                        E[e.to][e.rev].cap+=f;
                        ans+=(e.cost*f);
                        now=way_p[now];
      }
      return;
}                    
     
     
             
     
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(int i=0;i<n;i++){
            for(int j=0;j<i+m;j++){
                    scanf("%d",&num[i][j]);
            }
    }
    build_E();
    while(spfa()){work();}
    cout<<-ans<<endl;
    
    return 0;
}