【DP练习题】数字三角形3

题目：一个数字三角宝塔。

设数字三角形中的数字为绝对值不超过1000的整数。

小K从最顶层走到最底层，每一步可沿向下或右斜线向下走。

每走过一个节点他会把这个节点的数字加在自己计数器中。

另外他有一次机会，将他的计数器的数清零，他可以在任意时刻使用这次机会。

现在小K想知道他到达底层后，计数器中可能的最大的值.

输入数据 1

4
-1
6  5
4 -2 -5
1 11  6 20

输出数据 1

21

    本题与上篇数字三角形2区别就是在执行过程中可以使用一次清零机会，相对来说难度加大不少。看了某博主一些思路，可惜不是VIP看不全，经过多次测试终于AC。现将思路免费分享:

    核心思想：可以考虑用2个数组（就不用3维DP了）来记录每步状态，重要的是reset 清零DP数组，从最初就考虑清0的状态，或者说每步都执行清0，来比较与不清零DP最大值。因为会在每一步都在比较两个数组中的是大值，如果在某一步执行了清0，且值最大，那么在后续的累加中也一定最大，所以就不会再次出现需要清0的情况了。符合题意中只有一次执行清0的机会。同时与数字三角形2做法不同的是，这次从上到下走的方式，要注意数组边界的处理。

AC代码：

     基本上每行代码都有备注，应该好理解。

#include<bits/stdc++.h>
using namespace std;
const int INF = INT_MIN; // 用负无穷大表示不可达状态，即处理边界

int main() {
    int n; // 三角形的行数
    cin >> n;
    vector<vector<int>> a(n, vector<int>(n)); // 存储数字三角形的数组
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j <= i; ++j) {
            cin >> a[i][j];
        }
    }
    vector<vector<int>> dp(n, vector<int>(n, INF)); // 动态规划数组，dp[i][j]表示到达(i, j)时的最大值
    vector<vector<int>> reset(n, vector<int>(n, INF)); // reset[i][j]表示到达(i, j)时使用清零的最大值
    // 初始化第一行
    for (int j = 0; j < n; ++j) {
        dp[0][j] = a[0][j]; //初始化不清零数组的起点为原始值
        reset[0][j] = 0;  //初始化清零数组起始点为0
    }
    // 动态规划过程，每步都取两个数组中能往下和右下走的最大值
    for (int i = 1; i < n; ++i) {
        for (int j = 0; j <= i; ++j) {
            // 向下走
            if (j == 0) { // 如果是第一列，只能从上边下来
                dp[i][j] = dp[i-1][j] + a[i][j];
                reset[i][j] = max(reset[i-1][j], dp[i-1][j]) + a[i][j]; // 要么之前就用了清零，要么现在用
            } else if (j == i) { // 如果是最后一列，只能从左上角下来
                dp[i][j] = dp[i-1][j-1] + a[i][j];
                reset[i][j] = max(reset[i-1][j-1], dp[i-1][j-1]) + a[i][j];
            } else { // 中间列，可以从上边或左上角下来
                dp[i][j] = max(dp[i-1][j], dp[i-1][j-1]) + a[i][j];
                reset[i][j] = max({reset[i-1][j], reset[i-1][j-1], dp[i-1][j], dp[i-1][j-1]}) + a[i][j];
            }
// reset[i][j]还需要考虑当前位置清零的情况,即：使用清零机会时的值可能还不如不使用的值（即当前值可能是负的）)
            reset[i][j] = max(reset[i][j], a[i][j]); //在当前位置清零
        }
    }
    // 最后一行中的最大值即为所求
    int max1 = *max_element(dp[n-1].begin(), dp[n-1].end());
    int max2 = *max_element(reset[n-1].begin(), reset[n-1].end());
 	cout<<(max(max1,max2));
    return 0;
}