题目大意:
给定一个由n行数字组成的数字梯形如下图所示。梯形的第一行有m个数字。从梯形的顶部的m个数字开始,在每个数字处可以沿左下或者右下方向移动,形成一条从梯形的顶至底的路径。
规则1:从梯形的顶至底的m条路径互不相交。
规则2:从梯形的顶至底的m条路径仅在数字结合处相交。
规则3:从梯形的顶至底的m条路径允许在数字结合处相交或边相交。
对于给定的数字梯形,分别按照规则1,2,3,计算出从梯形顶至底的m条路径,使得这个m条路径经过的数字总和最大。
输入:
文件的第一行中有两个正整数m和n(m,n<=20),分别表示数字梯形的第一行有m个数字,共有n行。接下来n行是数字梯形中个行的数字。第一行有m个数字,第二行有m+1个数字。。。
思路分析:
规则1:
①:m条路径互不相交, 所以把每个点拆成<i,a>,<i,b>’两个点,从i向i‘连一条边,容量为1,费用为负的点权值,保证了路径上的每个点只经过一次;
②:设立一个源点s,从s向顶部每一个<i,a>连一条边,容量为1,费用为0;
③:设立一个汇点t,从底部每一个<i,b>向汇点连一条边,容量为1,费用为0;
④:对于每个点i向下层的两个点j,分别从<i,b>向<j,a>连一条边,容量为1,费用为0;
规则2:
m条路径可以在数字结点处相交,每个点可以经过多次,所以不需要拆点了;
①:从源点向顶部的每个点连一条边,容量为1,费用为0;
②:对于每个点i向下层的两个点j,分别从i向j连一条边,容量为1,费用为负的顶点i的权值;
③:从底部每一个i向汇点连一条边,容量为1,费用为负的顶点i的权值;
规则3:
规则3和规则2的建模方法第一条一样,把第二条第三条所建的边的容量改为INF就可以了。
代码实现:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#define Min(a,b) ((a)<(b)?(a):(b))
#define Max(a,b) ((a)>(b)?(a):(b))
using namespace std;
const int N=135200, M=4158000, INF=0x3f3f3f3f;
int n, m, s, t, top, tp, sum_cost;
int head[N], vis[N], dis[N], path[N], minflow[N], pre[N], num[45][45], cnt[45][45];
struct Edge{
int to,next,flow,cost;
Edge(int _to=0,int _next=0,int _flow=0,int _cost=0):to(_to),next(_next),flow(_flow),cost(_cost){}
}edge[M];
void Addedge(int from,int to,int flow,int cost){
edge[top] = Edge(to,head[from],flow,cost);
head[from] = top++;
edge[top] = Edge(from,head[to],0,-cost);
head[to] = top++;
}
int Spfa(){
queue<int> q;
memset(dis, 0x3f, sizeof(dis));
memset(minflow, 0x3f, sizeof(minflow));
memset(path, -1, sizeof(path));
memset(vis, 0, sizeof(vis));
dis[s]=0; q.push(s);
while(!q.empty()){
int u = q.front(); q.pop();
vis[u] = 0;
for(int i = head[u]; i + 1; i = edge[i].next){
if(edge[i].flow && dis[edge[i].to] > dis[u] + edge[i].cost){
dis[edge[i].to] = dis[u] + edge[i].cost;
pre[edge[i].to] = u;
path[edge[i].to] = i;
minflow[edge[i].to] = Min(minflow[u], edge[i].flow);
if(!vis[edge[i].to]){
vis[edge[i].to] = 1;
q.push(edge[i].to);
}
}
}
}
if(dis[t] == INF) return 0;
sum_cost += minflow[t]*dis[t];
int u = t;
while(u!=s){
edge[path[u]].flow -= minflow[t];
edge[path[u]^1].flow += minflow[t];
u = pre[u];
}
return 1;
}
int main(){
freopen("digit.in", "r", stdin);
freopen("digit.out", "w", stdout);
scanf("%d%d", &m, &n);
memset(head, -1, sizeof(head));
memset(cnt, 0, sizeof(cnt));
top = s = sum_cost = 0;
int c = 0;
for(int i = 1; i <= n; ++i){
for(int j = 1; j <= m+i-1; ++j){
cnt[i][j] = ++c;
scanf("%d", &num[i][j]);
}
}
t = c*2+1;
for(int i = 1; i <= n; ++i){
for(int j = 1; j <= m+i-1; ++j){
Addedge(cnt[i][j], cnt[i][j]+c, 1, -num[i][j]);
if(i == 1) Addedge(s, cnt[i][j], 1, 0);
if(i == n) Addedge(cnt[i][j]+c, t, 1, 0);
}
}
for(int i = 1; i < n; ++i){
for(int j = 1; j <= m+i-1; ++j){
Addedge(cnt[i][j]+c, cnt[i+1][j], 1, 0);
Addedge(cnt[i][j]+c, cnt[i+1][j+1], 1, 0);
}
}
while(Spfa());
printf("%d\n", -sum_cost);
memset(head, -1, sizeof(head));
top = sum_cost = 0;
for(int i = 1; i <= m; ++i) Addedge(s, i, 1, 0);
for(int i = 1; i <= m+n-1; ++i) Addedge(cnt[n][i], t, INF, -num[n][i]);
for(int i = 1; i < n; ++i){
for(int j = 1; j <= m+i-1; ++j){
Addedge(cnt[i][j], cnt[i+1][j], 1, -num[i][j]);
Addedge(cnt[i][j], cnt[i+1][j+1], 1, -num[i][j]);
}
}
while(Spfa());
printf("%d\n", -sum_cost);
memset(head, -1, sizeof(head));
top = sum_cost = 0;
for(int i = 1; i <= m; ++i) Addedge(s, i, 1, 0);
for(int i = 1; i <= m+n-1; ++i) Addedge(cnt[n][i], t, INF, -num[n][i]);
for(int i = 1; i < n; ++i){
for(int j = 1; j <= m+i-1; ++j){
Addedge(cnt[i][j], cnt[i+1][j], INF, -num[i][j]);
Addedge(cnt[i][j], cnt[i+1][j+1], INF, -num[i][j]);
}
}
while(Spfa());
printf("%d\n", -sum_cost);
}