[Leetcode]5982. 解决智力问题

【题目描述】

给你一个下标从 0 开始的二维整数数组 questions ，其中 questions[i] = [pointsi, brainpoweri] 。

这个数组表示一场考试里的一系列题目，你需要 按顺序 （也就是从问题 0 开始依次解决），针对每个问题选择 解决 或者 跳过 操作。解决问题 i 将让你 获得  pointsi 的分数，但是你将 无法 解决接下来的 brainpoweri 个问题（即只能跳过接下来的 brainpoweri 个问题）。如果你跳过问题 i ，你可以对下一个问题决定使用哪种操作。

比方说，给你 questions = [[3, 2], [4, 3], [4, 4], [2, 5]] ：
如果问题 0 被解决了， 那么你可以获得 3 分，但你不能解决问题 1 和 2 。
如果你跳过问题 0 ，且解决问题 1 ，你将获得 4 分但是不能解决问题 2 和 3 。
请你返回这场考试里你能获得的 最高 分数。

示例 1：

输入：questions = [[3,2],[4,3],[4,4],[2,5]]
输出：5
解释：解决问题 0 和 3 得到最高分。
- 解决问题 0 ：获得 3 分，但接下来 2 个问题都不能解决。
- 不能解决问题 1 和 2
- 解决问题 3 ：获得 2 分
总得分为：3 + 2 = 5 。没有别的办法获得 5 分或者多于 5 分。
示例 2：

输入：questions = [[1,1],[2,2],[3,3],[4,4],[5,5]]
输出：7
解释：解决问题 1 和 4 得到最高分。
- 跳过问题 0
- 解决问题 1 ：获得 2 分，但接下来 2 个问题都不能解决。
- 不能解决问题 2 和 3
- 解决问题 4 ：获得 5 分
总得分为：2 + 5 = 7 。没有别的办法获得 7 分或者多于 7 分。
 

提示：

1 <= questions.length <= 105
questions[i].length == 2
1 <= pointsi, brainpoweri <= 105

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/solving-questions-with-brainpower
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

【题目分析】

此题可以采用两种方法来做。

对于每一个问题，可以选择做或者不做，本质上是一个二叉树的遍历问题，规模为2的1e5的次方，所以直接深度搜索肯定会超时；

可以考虑采用备忘录进行优化，记录已经访问过的状态，使复杂度降到O(n);

我们可以定义DFS函数定义为从问题i出发所能获取的最大分数；

/*
    采用DFS + map进行优化；
    map[i]记录从i出发所获取的最高分数
*/
    unordered_map<int, long long> um;
    long long DFS(int i, vector<vector<int>>& questions, int n)
    {
        if (i >= n) {
            return 0;
        }
        if (um.find(i) != um.end()) {
            cout <<i << um[i] << endl;
            return um[i];
        }
        // 解决i
        long long score = questions[i][0];
        score += DFS(i + questions[i][1] + 1, questions, n);
        // 不解决
        long long res = max(score, DFS(i + 1, questions, n));
        um[i] = res;
        return res;
    }
    /*
    DFS与DFS2的区别是：
    DFS2将score带下去了，改变了状态，即不是从i出发的最大值了，而是加上了前面的分数；
    我们想要的状态是：从i出发所得到最大分数，即不能加上前面的状态；

    此题也可以用动态规划来求解，即从后往前；
    dp[n - 1] = ques[n - 1][0];
    dp[n - 2] = max(ques[n - 2][0], dp[n - 1])

    dp[i] = max(ques[i][0] + dp[i + ques[i][1] + 1], dp[i + 1])

    */
    long long DFS2(int i, vector<vector<int>>& questions, int n, long long score)
    {
        if (i >= n) {
            return score;
        }
        if (um.find(i) != um.end()) {
            cout <<i << um[i] << endl;
            return um[i];
        }
        // 解决i
        long long res = DFS2(i + questions[i][1] + 1, questions, n, score + questions[i][0]);
        // 不解决
        res = max(res, DFS2(i + 1, questions, n, score));
        um[i] = res;
        return res;
    }

DFS2的返回值中加上了之前的状态的结果，所以不是纯粹的从i出发所获取的最大值，因此结果是不对的；这种写法适合返回值为void的情况；

经过上面的分析可知，如果获得了从i出发的最大分数，那么就可以求得i - 1的最大分数，因此可以得到如下的递推关系：

    dp[n - 1] = ques[n - 1][0];

    dp[n - 2] = max(ques[n - 2][0], dp[n - 1])

        .....

    dp[i] = max(ques[i][0] + dp[i + ques[i][1] + 1], dp[i + 1])

 如果i + ques[i][1] + 1 > n ,那么 dp[i] = max(dp[i + 1], ques[i][0]);

代码如下：

    long long mostPoints(vector<vector<int>>& questions) {
        long long res = 0;

        int n = questions.size();
        int i = 0;
        long long score = 0;

        // res = DFS(0, questions, n);

        // 动态规划
        vector<long long>dp (n, 0);
        dp[n - 1] = questions[n - 1][0];
        for (int i = n - 2; i >= 0; i--) {
            if (i + questions[i][1] + 1 < n) {
                dp[i] = max(dp[i + 1], dp[questions[i][1] + i + 1] + questions[i][0]);
            } else {
                dp[i] = max(dp[i + 1], (long long)questions[i][0]);
            }
            
        }
        res = dp[0];
        return res;
    }