蓝桥杯第1场双周赛第4题 健身

解题思想 - 动态规划

1. 首先问题可以转换为:由q个不健身的日期(休息日)将n天划分为q+1个区间,求每个区间的最大健身奖励.最终所有区间的最大奖励和就是n天可获取的最大奖励.这个等价转换是一定正确的,是从题意获取的.
2. 所以要求出每个区间的最大奖励,联想到"凑零钱"问题. 决定采用dp做法
3. 如果每个区间都求一遍dp数组比较麻烦,不如直接一步求出dp数组.之后看区间多大就找dp数组中对应的结果即可

代码实现

import java.util.*;

public class LanQiao {

    public static long[] dp(int days, List<int[]> choices) {
        // dp[i]: i天的时间最多可以获得多少奖金
        long[] dp = new long[days + 1];

        // base case
        dp[0] = 0;

        //凑零钱的递推框架
        for (int i = 1; i <= days; i++) {
            for (int j = 0; j < choices.size(); j++) {
                int taskDay = choices.get(j)[0]; //一个多久的任务
                int taskBonus = choices.get(j)[1]; //该任务能获取多少奖励
                if (taskDay <= i) { //选择小于i天的任务,否则根本接不了
                    dp[i] = Math.max(dp[i - taskDay] + taskBonus, dp[i]); //转移方程,和凑零钱一摸一样
                } else { 
                    break; //因为对choices任务列表进行了排序,所以后面的任务都是大于i天的
                }
            }
        }
        return dp;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int q = sc.nextInt();

        // --- 求q+1个区间
        ArrayList<Integer> points = new ArrayList<>(q + 2);
        points.add(0);
        for (int i = 1; i <= q; i++) {
            points.add(sc.nextInt());
        }
        points.sort(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o1 - o2;
            }
        });
        points.add(n + 1);
        int[] intervals = new int[q + 1];
        for (int i = 0; i < q + 1; i++) {
            intervals[i] = points.get(i + 1) - points.get(i) -1;
        }
        // --- end

        // m个任务及其奖励
        List<int[]> tasks = new ArrayList<>(m);
        for (int i = 0; i < m; i++) {
            tasks.add(new int[]{(int)(Math.pow(2, sc.nextInt())), sc.nextInt()});
        }
        // 排序为了dp时的剪枝(见break)
        tasks.sort(new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return o1[0] - o2[0];
            }
        });

        // 一次性求出dp,留待使用
        long[] dp = dp(n, tasks);

        // 查询每一个区间天数对应的最大奖励并累加
        long sum = 0;
        for (int i = 0; i < q + 1; i++) {
            sum += dp[intervals[i]];
        }

        System.out.println(sum);
    }

代码细节

1. 求q+1个区间使用了小技巧, 阅读代码并结合下图体会:
   
2. 测试数据说明给得太少, 比如q个休息日给的顺序是升序的,所以可以不用对points排序就可以直接当作升序来用; m个任务不是有序的, 必须排序后才能使用break剪枝,否则结果错误.
3. 注意数据范围
4. 查看其它提交结果,发现一个dp写法不一样的, 我也没见过该写法, 先记录下来未来留意

    /**
     * @param dayDis 天数
     * @param w 任务的天数数组
     * @param v 任务的奖励数组
     * @return dayDis天数对应的最大奖励
     */
    static long dp(int dayDis, int w[], int v[]) {
        if (dayDis == 0)
            return 0;

        long dp[] = new long[dayDis + 1];

        // @TODO 这种状态转移的方式没见过
        for (int y = 0; y < w.length; ++y) {
            for (int i = w[y]; i <= dayDis; ++i) {
                dp[i] = Math.max(dp[i], dp[i - w[y]] + v[y]);
            }
        }
        return dp[dayDis];
    }

题目

链接

蓝桥杯

问题描述

输入格式

输出格式

样例输入

10 3 3
1 4 9
0 3
1 7
2 20

样例输出

30