Skip to content

【动规】解决智力问题

解决智力问题

给你一个下标从 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 。 请你返回这场考试里你能获得的 最高 分数。

记忆化搜索

比较容易想到的做法。

class Solution {
public:
    long long mostPoints(vector<vector<int>>& questions) {
        map<int, long long> m;
        function<long long(int)> solve = [&](int x) {
            if (m.count(x)) return m[x];
            if (x >= questions.size()) return (long long)0;
            long long a = solve(x + 1);
            long long b = questions[x][0] + solve(x + 1 + questions[x][1]);
            m[x] = max(a, b);
            return m[x];
        };
        return solve(0);
    }
};

倒序DP

\(f(i)\)为从第\(i\)题开始做的最高得分。由于用到之后的信息,需要倒序计算。

\[ \displaylines{ f(i) = \max\{f(i+1), \text{points}(i)+f(i+1+\text{brainpower}(i))\} } \]
class Solution {
public:
    long long mostPoints(vector<vector<int>>& questions) {
        int n = questions.size();
        vector<long long> dp(n + 1, 0);
        for (int i = n - 1; i >= 0; --i) {
            dp[i] = max(dp[i + 1], questions[i][0] + dp[min(n, i + questions[i][1] + 1)]);
        }
        return dp[0];
    }
};

正序DP

\(f(i)\)为前\(i\)题的最高得分。每次循环不只更新当前下标,还要更新未来的下标。

class Solution {
public:
    long long mostPoints(vector<vector<int>> &questions) {
        int n = questions.size();
        vector<long long> dp(n + 1);
        for (int i = 0; i < n; ++i) {
            dp[i + 1] = max(dp[i + 1], dp[i]);
            int j = min(n, i + questions[i][1] + 1);
            dp[j] = max(dp[j], dp[i] + questions[i][0]);
        }
        return dp[n];
    }
};