力扣 1696. 跳跃游戏 VI

1696. 跳跃游戏 VI

题目：

给你一个下标从 0 开始的整数数组 nums 和一个整数 k 。

一开始你在下标 0 处。每一步，你最多可以往前跳 k 步，但你不能跳出数组的边界。也就是说，你可以从下标 i 跳到 [i + 1， min(n - 1, i + k)] 包含 两个端点的任意位置。

你的目标是到达数组最后一个位置（下标为 n - 1 ），你的 得分 为经过的所有数字之和。

请你返回你能得到的 最大得分 。

示例 1：

输入：nums = [1,-1,-2,4,-7,3], k = 2
输出：7
解释：你可以选择子序列 [1,-1,4,3] （上面加粗的数字），和为 7 。
示例 2：

输入：nums = [10,-5,-2,4,0,3], k = 3
输出：17
解释：你可以选择子序列 [10,4,3] （上面加粗数字），和为 17 。
示例 3：

输入：nums = [1,-5,-20,4,-1,3,-6,-3], k = 2
输出：0

提示：

 1 <= nums.length, k <= 105
-104 <= nums[i] <= 104

方法一：动态规划+单调队列（转换为求滑动窗口最大值问题）

不断更新每个点位的值，利用单调队列找到跳点

class Solution {
public:
    int maxResult(vector<int>& nums, int k) 
    {
        int n=nums.size();
        deque<int> q;
        vector<int> sum(n);
        for(int i=0;i<n;i++)
        sum[i]=nums[i];//队列只保存下标值，sum[i]初始化为nums[i]的值方便取元素计算
        q.push_back(0);
        for(int i=1;i<n;i++)
        {
            while(q.size()&&i-q.front()>k)//去除队头合法元素，保持跳跃距离
            q.pop_front();
            sum[i]=sum[q.front()]+nums[i];//计算答案，即前面的跳点总和加上当前位置
            cout<<sum[i]<<endl;
            while(q.size()&&sum[i]>sum[q.back()])//不断寻找插入位置
            q.pop_back();
            q.push_back(i);//插入
        }
        return sum[n-1];
    }
};

方法二：动态规划+优先队列

class Solution {
public:
    int maxResult(vector<int>& nums, int k) {
        int n = nums.size();
        
        // 优先队列中的二元组即为 (f[j], j)
        priority_queue<pair<int, int>> q;
        q.emplace(nums[0], 0);
        int ans = nums[0];
        
        for (int i = 1; i < n; ++i){
            // 堆顶的 j 不满足限制
            while (i - q.top().second > k) {
                q.pop();
            }
            ans = q.top().first + nums[i];
            q.emplace(ans, i);
        }
        return ans;
    }
};

下一篇 快速幂算法