线性规划与网络流24——数字梯形问题

题目链接点这里

类似与最大权不相交路径，，但已经定下了S，和T，就简单多了

#include<iostream>
#include<map>
#include<stdio.h>
#include<algorithm>
#include<queue>
#include<set>
#include<string.h>
#include<string>
#include<vector>
#define MX 11111
#define INF 0x3f3f3f3f
#define mem(x,y) memset(x,y,sizeof(x))
#define FIN freopen("input.txt","r",stdin)
using namespace std;
int n,m,k;
int s,t;
int head[MX],rear;
struct Edge
{
    int nxt,to,cap,cost;
} edge[MX*11];
void edge_init()
{
    mem(head,-1);
    rear=0;
}
void edge_add(int a,int b,int cap,int cost)
{
    //cout<<a<<" "<<b<<endl;
    edge[rear].cost=cost;
    edge[rear].cap=cap;
    edge[rear].nxt=head[a];
    edge[rear].to=b;
    head[a]=rear++;
}
int dis[MX],pre[MX]; //储存前驱顶点
bool vis[MX];
bool spfa(int s, int t)
{
    mem(dis,0x3f);
    mem(vis,0);
    mem(pre,-1);
    deque<int>q;
    q.push_back(s);
    dis[s]=0,vis[s]=1;
    while (!q.empty())
    {
        int u = q.front();
        q.pop_front();
        vis[u] = 0;
        for (int i= head[u]; ~i; i=edge[i].nxt)
        {
            int v=edge[i].to;
            if (edge[i].cap&& dis[v]>dis[u]+edge[i].cost)
            {
                dis[v]=dis[u]+edge[i].cost;
                pre[v] = i;
                if (!vis[v])
                {
                    vis[v]=1;
                    if(!q.empty()&&dis[v]<=dis[q.front()])q.push_front(v);//SLF优化
                    else q.push_back(v);
                }
            }
        }
    }
    if (pre[t]==-1) return 0;
    else return 1;
}
//返回的是最大流， cost存的是最小费用
int MCMF(int s,int t,int &cost)
{
    int flow = 0;
    cost = 0;
    while (spfa(s, t))
    {

        int Min = INF;
        for (int i = pre[t]; i != -1; i = pre[edge[i ^ 1].to])
            if (Min > edge[i].cap) Min = edge[i].cap;
        for (int i = pre[t]; i != -1; i = pre[edge[i ^ 1].to])
            edge[i].cap -= Min,edge[i ^ 1].cap += Min;
        cost += dis[t]* Min;
        flow+=Min;
    }
    return flow;
}
int arr[111][111];
int lab[111][111];
int main()
{
    FIN;
    while(~scanf("%d%d",&m,&n))
    {
        int s=0,t=(m+m+n-1)*n+1,w=(m+m+n-1)*n/2;

        int cnt=0;
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=m+i-1; j++)
            {
                scanf("%d",&arr[i][j]);
                lab[i][j]=++cnt;
            }
        }

        edge_init();
        for(int i=1; i<=m; i++)edge_add(s,lab[1][i],1,0),edge_add(lab[1][i],s,0,0);
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=m+i-1; j++)
            {
                edge_add(lab[i][j],w+lab[i][j],1,-arr[i][j]),edge_add(w+lab[i][j],lab[i][j],0,arr[i][j]);
                if(i!=n)
                {
                    edge_add(w+lab[i][j],lab[i+1][j],1,0),edge_add(lab[i+1][j],w+lab[i][j],0,0);
                    edge_add(w+lab[i][j],lab[i+1][j+1],1,0),edge_add(lab[i+1][j+1],w+lab[i][j],0,0);
                }
                else edge_add(w+lab[i][j],t,1,0),edge_add(t,w+lab[i][j],0,0);
            }
        }
        int cost;
        MCMF(s,t,cost);
        printf("%d\n",-cost);
        edge_init();
        for(int i=1; i<=m; i++)edge_add(s,lab[1][i],1,0),edge_add(lab[1][i],s,0,0);
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=m+i-1; j++)
            {
                edge_add(lab[i][j],w+lab[i][j],INF,-arr[i][j]),edge_add(w+lab[i][j],lab[i][j],0,arr[i][j]);
                if(i!=n)
                {
                    edge_add(w+lab[i][j],lab[i+1][j],1,0),edge_add(lab[i+1][j],w+lab[i][j],0,0);
                    edge_add(w+lab[i][j],lab[i+1][j+1],1,0),edge_add(lab[i+1][j+1],w+lab[i][j],0,0);
                }
                else edge_add(w+lab[i][j],t,INF,0),edge_add(t,w+lab[i][j],0,0);
            }
        }

        cost;
        MCMF(s,t,cost);
        printf("%d\n",-cost);
        edge_init();
        for(int i=1; i<=m; i++)edge_add(s,lab[1][i],1,0),edge_add(lab[1][i],s,0,0);
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=m+i-1; j++)
            {
                edge_add(lab[i][j],w+lab[i][j],INF,-arr[i][j]),edge_add(w+lab[i][j],lab[i][j],0,arr[i][j]);
                if(i!=n)
                {
                    edge_add(w+lab[i][j],lab[i+1][j],INF,0),edge_add(lab[i+1][j],w+lab[i][j],0,0);
                    edge_add(w+lab[i][j],lab[i+1][j+1],INF,0),edge_add(lab[i+1][j+1],w+lab[i][j],0,0);
                }
                else edge_add(w+lab[i][j],t,INF,0),edge_add(t,w+lab[i][j],0,0);
            }
        }
        cost;
        MCMF(s,t,cost);
        printf("%d\n",-cost);
    }
    return 0;
}