【洛谷】2990:[USACO10OPEN]牛跳房子Cow Hopscotch【单调队列优化DP】
题目描述
The cows have reverted to their childhood and are playing a game similar to human hopscotch. Their hopscotch game features a line of N (3 <= N <= 250,000) squares conveniently labeled 1..N that are chalked onto the grass.
Like any good game, this version of hopscotch has prizes! Square i is labeled with some integer monetary value V_i (-2,000,000,000 <= V_i <= 2,000,000,000). The cows play the game to see who can earn the most money.
The rules are fairly simple:
* A cow starts at square '0' (located just before square 1; it has no monetary value).
* She then executes a potentially empty sequence of jumps toward square N. Each square she lands on can be a maximum of K (2 <= K <= N) squares from its predecessor square (i.e., from square 1, she can jump outbound to squares 2 or 3 if K==2).
* Whenever she wishes, the cow turns around and jumps back towards square 0, stopping when she arrives there. In addition to the restrictions above (including the K limit), two
additional restrictions apply:
* She is not allowed to land on any square she touched on her outbound trip (except square 0, of course).
* Except for square 0, the squares she lands on during the return trip must directly precede squares she landed on
during the outbound trip (though she might make some larger leaps that skip potential return squares altogether).
She earns an amount of money equal to the sum of the monetary values of all the squares she jumped on. Find the largest amount of cash a cow can earn.
By way of example, consider this six-box cow-hopscotch course where K has the value 3:
Square Num: 0 1 2 3 4 5 6
+---+ +---+ +---+ +---+ +---+ +---+ +---+
|///|--| |--| |--| |--| |--| |--| |
+---+
+---+
+-
--+
+--
-+
+--
-+
+---
+
+
---+
Val
ue:
- 0 1 2 -3 4 5
One (optimal) sequence Bessie could jump (shown with respective bracketed monetary values) is: 1[0], 3[2], 6[5], 5[4], 2[1], 0[0] would yield a monetary total of 0+2+5+4+1+0=12.
If Bessie jumped a sequence beginning with 0, 1, 2, 3, 4, ... then she would be unable to return since she could not legally jump back to an untouched square.
奶牛们正在回味童年,玩一个类似跳格子的游戏,在这个游戏里,奶
牛们在草地上画了一行N个格子,(3 <=N <= 250,000),编号为1..N。
就像任何一个好游戏一样,这样的跳格子游戏也有奖励!第i个格子标有一个数字V_i(-2,000,000,000 <=V_i <= 2,000,000,000)表示这个格子的钱。奶牛们想看看最后谁能得到最多的钱。
规则很简单:
* 每个奶牛从0号格子出发。(0号格子在1号之前,那里没钱)
* 她向N号格子进行一系列的跳跃(也可以不跳),每次她跳到的格子最多可以和前一个落脚的格子差K格(1 <= K <= N)(比方说,当前在1号格,K=2, 可以跳到2号和3号格子)
*在任何时候,她都可以选择回头往0号格子跳,直到跳到0号格子。另外,除了以上规则之外,
回头跳的时候还有两条规则:
*不可以跳到之前停留的格子。
*除了0号格子之外,她在回来的时候,停留的格子必须是恰巧过去的时候停留的某个格子的前一格(当然,也可以跳过某些过去…
输入输出格式
输入格式:
* Line 1: Two space separated integers: N and K
* Lines 2..N+1: Line i+1 contains a single integer: V_i
输出格式:
* Line 1: A single line with a single integer that is the maximum amount of money a cow can earn
英文看着头大,实际上就是在坐标轴上跳,每次最多跳$k$步,还要跳回来,而跳回来时踩的点必须是跳过去时经过过的点的前一个,求这样跳过去跳回原点能获得的最大价值。
而很容易想到,因为回来时的路径选择是依赖过去的路径,我们可以DP过去的路径,同时为回来的路径预留位置,所以我们的DP方程$f[i]$表示过去时跳到$i$位置此时能获得的最大价值,转移就是从$f[i-k]-f[i-2]$中取max(就是选最优决策点),再加上选的最有决策点到当前位置$i$中区间正数的和(因为k距离以内可以随便跳,我们贪心选择这段区间价值大于0的跳就行了),最后加上$a[i]$和$a[i-1]$就行了。
因为要找的实际上就是在k范围内的最大$f[j]-pre[j]$,很容易想到用单调队列来优化。然后问题就解决了,主要是各种细节。
首先要保证队列中能更新$i$的决策点都是在$i-k$到$i-2$之间的,所以随着$i$的加入顺势加入当前新的决策点$j$就可以了。
优先队列一般都是先删掉劣的尾,加入当前的新值,再删过期的头,最后更新要求得的$dp$方程。删尾时要保证队列内不为空,删头时除过期的还要满足题目要求的限定条件,比如队列内必须至少存在几个元素。
这道题最后统计答案时不能忘了,一步跳过去和一步跳回来也可能是一个答案,要先定$ans$初值。
#include<bits/stdc++.h> #define LL long long using namespace std; int n, k; int a[250005]; LL q[250005], pre[250005], f[250005]; int main() { scanf("%d%d", &n, &k); for(int i = 1; i <= n; i ++) { scanf("%d", &a[i]); pre[i] = pre[i-1] + (a[i] > 0 ? a[i] : 0); } int h = 0, t = 0; q[++t] = 1; f[1] = a[1]; for(int i = 2,j = 0; i <= n; i ++,j ++) {//因为队列中位置必须小于等于i-2才能更新i 所以定义j来直接表示 LL tmp = f[j] - pre[j]; while(t - h > 0 && tmp >= f[q[t]] - pre[q[t]]) t --; q[++t] = j; while(i - q[h+1] > k && t - h > 1) h ++; ///////是由i来判断队列中点是否过期 f[i] = f[q[h+1]] - pre[q[h+1]] + a[i] + a[i-1] + pre[i-2]; } LL ans = pre[min(n, k)];//跳k步过去再直接跳回来 for(int i = 1; i <= n; i ++) { int to = min(n, i - 1 + k); ans = max(ans, f[i] + pre[to] - pre[i]); } printf("%lld", ans); return 0; }