华为OD机试真题 Java 实现 [最大报酬][工作安排]

题目描述

工作安排
小明每周上班都会拿到自己的工作清单，工作清单内包含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的最大报酬
    }
}