[数塔问题] 题解&代码

题目看到一半，莫名有点小兴奋，dp第一水题——数字三角形。
结果，看到最后一行，我就缄默了。———— 题记

【问题描述】：

相信大家都写过数字三角形问题，题目很简单求最大化一个三角形数塔从上往下走的路径和。走的规则是：$（i，j）$号点只能走向$（i+1，j）$或者$（i+1，j+1）$。如下图是一个数塔，映射到该数塔上行走的规则为：从左上角的点开始，向下走或向右下走直到最底层结束。
$1$
$3$ $8$
$2$ $5$ $0$
$1$ $4$ $3$ $8$
$1$ $4$ $2$ $5$ $0$

路径最大和是$1+8+5+4+4=22$，$1+8+5+3+5=22$或者$1+8+0+8+5=22$。

小$S$觉得这个问题$so$ $easy$。于是他提高了点难度，他每次$ban$掉一个点（即规定哪个点不能经过），然后询问你不走该点的最大路径和。

当然他上一个询问被$ban$掉的点过一个询问会恢复（即每次他在原图的基础上$ban$掉一个点，而不是永久化的修改）。

【输入格式】：

第一行包括两个正整数，$N$，$M$，分别表示数塔的高和询问次数。
以下$N$行，第i行包括用空格隔开的$i-1$个数，描述一个高为N的数塔。
而后$M$行，每行包括两个数$X$，$Y$，表示第$X$行第$Y$列的数塔上的点被小$S$ $ban$掉，无法通行。
（由于读入数据较大，$c$或$c++$请使用较为快速的读入方式）

【输出格式】：

$M$行每行包括一个非负整数，表示在原图的基础上$ban$掉一个点后的最大路径和，如果被$ban$掉后不存在任意一条路径，则输出$"-1"$。

【输入输出样例1】：

5 3
1
3 8
2 5 0
1 4 3 8
1 4 2 5 0
2 2
5 4
1 1 17
22
-1

【样例解释】：

第一次是

$1$
$3$ $X$
$2$ $5$ $0$
$1$ $4$ $3$ $8$
$1$ $4$ $2$ $5$ $0$
$1+3+5+4+4=17$ 或者 $1+3+5+3+5=17$

第二次是：

$1$
$3$ $X$
$2$ $5$ $0$
$1$ $4$ $3$ $8$
$1$ $4$ $2$ $X$ $0$
$1+8+5+4+4=22$
第三次：你们都懂的！无法通行，-1！

【数据规模及约定】：

---

解析：

这道题实际意思就是求不经过当前点的数字三角形最大值。

易证每一层只选一个值，所以我们就可以通过预处理，处理出$up$数组和$down$数组，$up$数组表示每个点向上走的最大$dp$值，同理，$down$数组表示每个点向下走的$dp$值，于是，我们就可以得到经过一个点的最大路径。

因为题目要求是求不经过当前点的数字三角形最大值，又因为每层只有一个点会通过（选择），我们预处理后，就可以预处理经过每个点的最大路径的最大值，以及不同于最大值位置的次大值。

这样，对于每次询问，如果这个点是该层最大路径的位置，就输出次大值，反之，就输出最大值。

时间复杂度：$O(n^2+m).$

预计得分：100

代码：

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

const int N = 2001;
int n, m;
int a[N][N], up[N][N], down[N][N];
int Max1[N], Max2[N], js[N];

int main()
{
	freopen("tower.in","r",stdin);
	freopen("tower.out","w",stdout);
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++) {
		for (int j=1;j<=i;j++) {
			scanf("%d",&a[i][j]), up[i][j] = down[i][j] = a[i][j];
		}
	}
	
	for (int i=n-1;i>=1;i--) {
		for (int j=1;j<=i;j++) {
			down[i][j] = a[i][j] + max(down[i+1][j], down[i+1][j+1]);
		}
	}
	for (int i=2;i<=n;i++) {
		for (int j=1;j<=i;j++) {
			up[i][j] = a[i][j] + max(up[i-1][j], up[i-1][j-1]);
		}
	}
	
	for (int i=1;i<=n;i++) {
		for (int j=1;j<=i;j++) {
			int sum = up[i][j] + down[i][j] - a[i][j];
	      	if (sum > Max1[i]) {
	      		Max2[i] = Max1[i];
	      		Max1[i] = sum;
	      		js[i] = j;
			}
			else Max2[i] = max(Max2[i], sum);
		}
	}
	for (int i=1;i<=m;i++) {
		int x, y;
		scanf("%d%d",&x,&y);
		if (x == 1 && y == 1) {
			puts("-1");
			continue;
		}
		if (js[x] == y) printf("%d\n",Max2[x]);
			else printf("%d\n",Max1[x]);
	}
	return 0;
}