【Lintcode】273. Test Strategy

题目地址：

https://www.lintcode.com/problem/test-strategy/description

有一场考试，总时间$120$分钟，给定四个数组p，part，f，full，分别表示每道题如果只做一部分花的时间、得到的分数，和每道题做完整花的时间、得到的分数。问能得到的最大分数是多少。

思路是动态规划。设$g[i][t]$是如果只做第$0$到第$i$道题，在$t$这么多时间内，能得到的最大分数。那么按照第$i$道题怎么做，可以分为三类，一类是不做，一类是做部分，一类是做完整，则有： g [ i ] [ t ] = max ⁡ { g [ i − 1 ] [ t ] , g [ i − 1 ] [ t − p [ i ] ] + p a r t [ i ] , g [ i − 1 ] [ t − f [ i ] ] + f u l l [ i ] } g[i][t]=\max\{g[i-1][t],g[i-1][t-p[i]]+part[i],g[i-1][t-f[i]]+full[i]\} g[i][t]=max{g[i−1][t],g[i−1][t−p[i]]+part[i],g[i−1][t−f[i]]+full[i]}初始条件是要求$g[0][t]$，也可以分为上面三类，详细解释这里省略。代码如下：

public class Solution {
    /**
     * @param p:    The time you choose to do part of the problem.
     * @param part: The points you choose to do part of the problem.
     * @param f:    The time you choose to do the whole problem.
     * @param full: The points you choose to do the whole problem.
     * @return: Return the maximum number of points you can get.
     */
    public int exam(int[] p, int[] part, int[] f, int[] full) {
        // write your code here
        int n = p.length;
        int[][] dp = new int[n][121];
        // 枚举第0道题怎么处理
        for (int i = p[0]; i <= 120; i++) {
            dp[0][i] = part[0];
            if (i >= f[0]) {
                dp[0][i] = full[0];
            }
        }
        
        for (int i = 1; i < n; i++) {
            for (int j = 0; j <= 120; j++) {
            	// 接下来枚举第i题怎么处理。如果第i题不做，那答案就是dp[i - 1][j]
                dp[i][j] = dp[i - 1][j];
                // 枚举只做部分的情况
                if (j >= p[i]) {
                    dp[i][j] = Math.max(dp[i][j], dp[i - 1][j - p[i]] + part[i]);
                }
                // 枚举做完整的情况
                if (j >= f[i]) {
                    dp[i][j] = Math.max(dp[i][j], dp[i - 1][j - f[i]] + full[i]);
                }
            }
        }
        
        return dp[n - 1][120];
    }
}

时空复杂度$O(n)$，$n$是题目数量。