【网络流24题】数字梯形问题 题解

题目传送门

题目大意： 给出一个数字梯形，顶有 $n$ 个数字，高度为 $m$，从顶部每一个数字出发，可以向左下或右下走，一直走到底，一条路径的值定义为路上经过的所有数字之和。给出 $3$ 个条件，分别在 $3$ 个条件下求出 $n$ 条路径的最大值。条件1：$m$ 条路径互不相交；条件2：$m$ 条路径可以在点上相交；条件3：$m$ 条路径可以在点和边上相交。

题解

裸的费用流，只需要根据条件调整费用即可。

总的建图：源点向顶的每一个数建，流量为 $1$，费用为 $0$，然后就是每个数字向左下和右下建，流量为 $1$，费用为 $0$，以及底向汇点建，流量为 $1$，费用为 $0$，由于有路径不能相交，即每个点只能走1次这样的限制，所以每个点要拆点，中间连一条流量为 $1$，费用为数字的边。

这是条件1的，条件2的话只需要将拆点中间的边流量改成无限，以及底向汇点的流量无限即可。

条件3的话在条件2基础上把每个数字向左下和右下建的边流量改成无限即可。

一开始加了点很蠢的判断搞得我调了20min……

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 100010
#define P (n+m)*(n+m)
#define inf 999999999

int n,m,S,T,a[110][110],id=0,pos[110][110];
struct edge{int x,y,z,cost,next;};
edge e[maxn<<1];
int first[maxn],len;
void buildroad(int x,int y,int z,int cost)
{
	e[++len]=(edge){x,y,z,cost,first[x]};
	first[x]=len;
}
int q[maxn],st,ed,f[maxn],fa[maxn];
bool v[maxn];
bool SPFA()
{
	for(int i=1;i<=T;i++)f[i]=-inf;
	memset(fa,0,sizeof(fa));
	st=1;ed=2;q[st]=S;v[S]=true;f[S]=0;
	while(st!=ed)
	{
		int x=q[st++];st=st>T?1:st;v[x]=false;
		for(int i=first[x];i;i=e[i].next)
		{
			int y=e[i].y;
			if(e[i].z&&f[y]<f[x]+e[i].cost)
			{
				f[y]=f[x]+e[i].cost;fa[y]=i;
				if(!v[y])v[q[ed++]=y]=true,ed=ed>T?1:ed;
			}
		}
	}
	return fa[T];
}
void work(int x,int y)
{
	memset(first,0,sizeof(first));len=1;
	for(int i=1;i<=n;i++)//源点到第一排 
	buildroad(S,pos[1][i],1,0),buildroad(pos[1][i],S,0,0);
	for(int i=1;i<n+m;i++)//最后一排到汇点 
	buildroad(pos[m][i]+P,T,x,0),buildroad(T,pos[m][i]+P,0,0);
	for(int i=1;i<m;i++)//向下走的边 
	for(int j=1;j<n+i;j++)
	buildroad(pos[i][j]+P,pos[i+1][j],y,0),buildroad(pos[i+1][j],pos[i][j]+P,0,0),
	buildroad(pos[i][j]+P,pos[i+1][j+1],y,0),buildroad(pos[i+1][j+1],pos[i][j]+P,0,0);
	for(int i=1;i<=m;i++)//拆开的两点建边 
	for(int j=1;j<n+i;j++)
	buildroad(pos[i][j],pos[i][j]+P,x,a[i][j]),buildroad(pos[i][j]+P,pos[i][j],0,-a[i][j]);
	int ans=0;
	while(SPFA())for(int i=fa[T];i;i=fa[e[i].x])e[i].z--,e[i^1].z++,ans+=e[i].cost;
	printf("%d\n",ans);
}

int main()
{
	scanf("%d %d",&n,&m);S=2*P+1;T=S+1;
	for(int i=1;i<=m;i++)
	for(int j=1;j<n+i;j++)
	scanf("%d",&a[i][j]),id++,pos[i][j]=id;
	work(1,1);work(inf,1);work(inf,inf);
}