「网络流 24 题」数字梯形

#include<bits/stdc++.h>
using namespace std;

long long tot=-1,sum=0,h[20005],flow[20005],g[20005],ans=0,dis[20005],ans2=0,inf=9999999;
bool vis[20005];
struct node{
    long long from,to,next,rest,cost;
}e[10000005];

void add(long long x,long long y,long long z,long long hg){
    tot++;
    e[tot].next=h[x];
    h[x]=tot;
    e[tot].cost=hg;
    e[tot].from=x;
    e[tot].to=y;
    e[tot].rest=z;
}

int bfs(long long s,long long t){
    queue<int>q;
    q.push(s);dis[s]=0;vis[s]=true;
    while(!q.empty()){
        long long u=q.front();q.pop();//vis[u]=false;
        for(int i=h[u];i!=(-1);i=e[i].next){
            if(e[i].rest>0&&dis[e[i].to]<dis[u]+e[i].cost){
                flow[e[i].to]=min(flow[u],e[i].rest);
                dis[e[i].to]=dis[u]+e[i].cost;
                g[e[i].to]=i;
                if(vis[e[i].to]==false)q.push(e[i].to),vis[e[i].to]=true;
            }
        }
    }
}

int EK(long long s,long long t){
    while(1){
        memset(flow,0x7f,sizeof(flow));
        memset(dis,-0x7f,sizeof(dis));
        memset(g,-1,sizeof(g));
        memset(vis,false,sizeof(vis));
        bfs(s,t);
        if(g[t]!=(-1)){
            ans+=flow[t];ans2+=dis[t]*flow[t];
            for(int p=t;p!=s;p=e[g[p]].from){
                e[g[p]].rest-=flow[t];
                e[g[p]^1].rest+=flow[t];
            }
        }
        else break; 
    }
}

void init(){
    memset(h,-1,sizeof(h));
    tot=-1;
    ans=0;
    ans2=0;
}

int m,n,tu[2005][2005],hj=0,zb[2005][2005];

int main(){
    cin>>m>>n;
    for(int i=1;i<=n;i++)for(int j=1;j<m+i;j++){
    cin>>tu[i][j];hj++;
    zb[i][j]=hj;    
    }
    //塑造管道
    init();
    for(int i=1;i<=n;i++){
        for(int j=1;j<m+i;j++){
            add(zb[i][j],zb[i][j]+5000,1,tu[i][j]);
            add(zb[i][j]+5000,zb[i][j],0,-tu[i][j]);
            if(i==1){
                add(0,zb[i][j],1,0);
                add(zb[i][j],0,0,0);
            }
            if(i!=n){
                add(zb[i][j]+5000,zb[i+1][j],1,0);
                add(zb[i+1][j],zb[i][j]+5000,0,0);
                add(zb[i][j]+5000,zb[i+1][j+1],1,0);
                add(zb[i+1][j+1],zb[i][j]+5000,0,0);
            }
            if(i==n){
                add(zb[i][j]+5000,15000,1,0);
                add(15000,zb[i][j]+5000,0,0);
            }
        }
    }
    EK(0,15000);
    cout<<ans2<<endl;
    //第一问,直接根据点的关系建边 
    init();
    for(int i=1;i<=n;i++){
        for(int j=1;j<m+i;j++){
            if(i==1){
                add(0,zb[i][j],1,0);
                add(zb[i][j],0,0,0);
            }
            if(i!=n){
                add(zb[i][j],zb[i+1][j],1,tu[i][j]);
                add(zb[i+1][j],zb[i][j],0,-tu[i][j]);
                add(zb[i][j],zb[i+1][j+1],1,tu[i][j]);
                add(zb[i+1][j+1],zb[i][j],0,-tu[i][j]);
            }
            if(i==n){
                add(zb[i][j],9999,inf,tu[i][j]);
                add(9999,zb[i][j],0,-tu[i][j]);
            }
        }
    }
    EK(0,9999);
    cout<<ans2<<endl; 
    //第二问,点可以相交,那么边就一定不可以相加,那么可以把 边化作点,点化作边 
    init();
    for(int i=1;i<=n;i++){
        for(int j=1;j<m+i;j++){
            if(i==1){
                add(0,zb[i][j],1,tu[i][j]);
                add(zb[i][j],0,0,-tu[i][j]);
            }
            if(i!=n){
                add(zb[i][j],zb[i+1][j],inf,tu[i+1][j]);
                add(zb[i+1][j],zb[i][j],0,-tu[i+1][j]);
                add(zb[i][j],zb[i+1][j+1],inf,tu[i+1][j+1]);
                add(zb[i+1][j+1],zb[i][j],0,-tu[i+1][j+1]);
            }
            if(i==n){
                add(zb[i][j],n*m+1,inf,0);
                add(n*m+1,zb[i][j],0,0);
            }
        }
    }
    EK(0,n*m+1);
    cout<<ans2<<endl;
    //第三问 点和边都可以相交... 没有限制了直接全部一起跑只是起点不同而以
     
}