目录
题目
题目描述
有Q只猴子要从第一棵树到第n棵树去,第i只猴子一次跳跃的最远距离为Ki。如果它在第x棵树,那它最远可以跳到第x+Ki棵树。如果第j棵树的高度比第i棵树高或相等,那么它从第i棵树直接跳到第j棵树,它的劳累值会增加1。所有猴子一开始在第一棵树,请问每只猴子要跳到第n棵树花费的劳累值最小。
输入
第一行一个整数n,表示有n棵树。(2<=n<=1000000),接下来第二行给出n个正整数D1,D2,……,Dn(1<=Di<=10^9),其中Di表示第i棵树的高度。第三行给出了一个整数Q(1<=Q<=25),接下来Q行,给出了每只猴子一次跳跃的最远距离Ki(1<=Ki<=N-1)。
输出
输出Q行,每行一个整数,表示一只猴子的最小的劳累值。
样例输入
9
4 6 3 6 3 7 2 6 5
2
2
5
样例输出
2
1
解题思路
可以看到,这里所有的题都是dp,所以,这里就只考虑dp怎么实现正解。
可以看到,令dp[i] 是为到第i棵树时前面最小的疲劳值,转移方程可以很快地得到:dp[i] = min (dp[k] + (h[k] <= h[i]));
这里的h[k] <= h[i] 也就是说如果第k棵树比第i棵树矮或相等的话就会加1(详见题目)。
然后k的取值也是i - K[i] <= k < i,也就是往前枚举。或许会有人说了,这样的时间复杂度不就是 O ()吗,这样子铁定超时啊。
别慌别慌,这不是就到主题了吗——用单调栈、队列来优化dp。
我们可以用单调上升的队列来存储从i - K[i] ~ i 之间的最小值,把它赋给dp[i],再把dp[i]放到合适的位置,不就万事大吉了吗!
当然,这里还有一个小问题,将dp[i]放到合适的位置时,如果dp[i]大于了队尾,那么就直接放到它后面就行了;如果是小于,就一直到比它还小的元素后面;
可是如果是等于呢?
仔细想想可以想到,越高的树是越有潜力的,在后面的作用更大。所以如果是队尾的高度小于了该高度,那么也直接踢掉,否则就加到后面去。
代码
#include <cstdio>
#include <deque>
#include <cstring>
using namespace std;
#define MAXN 1000000
#define INF 0x3f3f3f3f
int n, d[MAXN + 5], q, k[MAXN + 5], dp[MAXN + 5];
deque <int> D;
void Clear (){
memset (dp, INF, sizeof (dp));
D.clear ();
}
void read (int &x){
x = 0;
char c = getchar ();
while (c < '0' || c > '9')
c = getchar ();
while (c >= '0' && c <= '9'){
x = (x << 1) + (x << 3) + c - 48;
c = getchar ();
}
}
void print (int x){
if (x < 0){
putchar ('-');
x = (~ x) + 1;
}
if (x / 10)
print (x / 10);
putchar (x % 10 + 48);
}
int main (){
read (n);
for (int i = 1; i <= n; i++)
read (d[i]);
read (q);
for (int i = 1; i <= q; i++){
read (k[i]);
Clear ();
dp[1] = 0;
D.push_back (1);
for (int j = 2; j <= n; j++){
int Min = j - k[i];
if (Min < 1)
Min = 1;
while (!D.empty ()){
int x = D.front ();
D.pop_front ();
if (x < Min)
continue;
else{
D.push_front (x);
break;
}
}
int x = D.front ();
dp[j] = dp[x];
if (d[j] >= d[x])
dp[j] ++;
while (!D.empty ()){
int x = D.back ();
D.pop_back ();
if (dp[x] > dp[j])
continue;
else if (dp[x] == dp[j] && d[x] < d[j])
continue;
else{
D.push_back (x);
break;
}
}
D.push_back (j);
}
print (dp[n]);
putchar ('\n');
}
}