题目描述
工作安排
小明每周上班都会拿到自己的工作清单,工作清单内包含n项工作,每项工作都有对应的耗时时长(单位h)和报酬,工作的总报酬为所有已完成工作的报酬之和。那么请你帮小明安排一下工作,保证小明在指定的工作时间内工作收入最大化。
输入描述
输入的第一行为两个正整数T,n。
T代表工作时长(单位h,0<T<100000),n代表工作数量(1 < n< 3000)。
接下来是n行,每行包含两个整数t,w。
t代表该项工作消耗的时长(单位h,t>0),w代表该项工作的报酬。
输出描述
输出小明指定工作时长内工作可获得的最大报酬。
示例一
输入
40 3
20 10
20 20
20 5
输出
30
思路:
因为每个工作只能选择一次,所以这是一道 0-1 背包问题,使用动态规划算法求解。
使用一个二维数组 dp 存储状态,其中 dp[i][j] 表示前 i 个工作中,总工作时间不超过 j 的最大报酬。
对于第 i 个工作,如果它的耗时时长小于等于当前总工作时间 j,则可以考虑选择该工作,此时可以得到的最大报酬为 dp[i-1][j] 和 dp[i-1][j-works[i-1][0]]+works[i-1][1] 中的较大值;如果它的耗时时长大于当前总工作时间 j,则无法选择该工作,此时可以得到的最大报酬为 dp[i-1][j]。
这样就得到了状态转移方程:
dp[i][j] = max(dp[i-1][j], dp[i-1][j-works[i-1][0]]+works[i-1][1]) // j >= works[i-1][0]
dp[i][j] = dp[i-1][j] // j < works[i-1][0]
最终结果为 dp[n][T],即前 n 个工作中,总工作时间不超过 T 的最大报酬。
代码实现:
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int T = scanner.nextInt(); // 工作时长
int n = scanner.nextInt(); // 工作数量
int[][] works = new int[n][2]; // 存储工作信息,works[i][0]表示第i个工作的时长,works[i][1]表示第i个工作的报酬
for (int i = 0; i < n; i++) {
works[i][0] = scanner.nextInt();
works[i][1] = scanner.nextInt();
}
int[][] dp = new int[n + 1][T + 1]; // dp[i][j]表示前i个工作中,总工作时间不超过j的最大报酬
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= T; j++) {
if (j >= works[i - 1][0]) { // 如果当前工作的时长小于等于当前总工作时间j,则可以考虑选择当前工作
dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j - works[i - 1][0]] + works[i - 1][1]);
} else { // 如果当前工作的时长大于当前总工作时间j,则无法选择当前工作
dp[i][j] = dp[i - 1][j];
}
}
}
System.out.println(dp[n][T]); // 输出前n个工作中,总工作时间不超过T的最大报酬
}
}