题目:
给你一个下标从 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; } };