题意:
有nnn棵树,每棵高hih_{i}hi,有mmm只鸟,每只鸟有一个参数kik_{i}ki,代表这只鸟在xxx位置时能跳到[x,x+k][x,x+k][x,x+k]的任意一个位置。如果从iii跳往jjj,并且有hi<=hjh_{i}<=h_{j}hi<=hj,那么这次跳跃需要一个单位的劳累值。所有鸟都需要从111跳到nnn,问每只鸟的最小劳累值总和。
方法:
设dp[i]dp[i]dp[i]为到达iii的最小劳累值总和,那么dp[i]=min(dp[j]+(h[j]<=h[i])),j∈[max(1,i−k),i−1]dp[i]=min(dp[j]+(h[j]<=h[i])),j\in[max(1,i-k),i-1]dp[i]=min(dp[j]+(h[j]<=h[i])),j∈[max(1,i−k),i−1]
我们可以在按顺序计算dp[i]dp[i]dp[i]的同时,维护一个单调队列,来O(1)O(1)O(1)的获得[max(1,i−k),i−1][max(1,i-k),i-1][max(1,i−k),i−1]的最大值。
维护的不应该是最终对iii的贡献,应该维护的仅仅是dp[i]dp[i]dp[i],因为iii变化时h[i]h[i]h[i]变化,对这个位置的贡献可能发生变化。
那么维护得到了多个值相同的位置,应该是选择一个最优的:
一种思考是遍历所有有最小值的位置,我们取其中一个位置比现在高的,这样可以不用付出劳累值,如果不存在比现在高的,我们就用最小值+1,这样的决策一定不会更劣,应该最小值+1一定还是小于等于次小值的。这样的最坏时间复杂度是O(n2m)O(n^2m)O(n2m)
在这样的基础上贪心思考,我们如果在拥有同样最小值的位置里面只保存最高的地方,这一定是更优的,不需要全部遍历,复杂度是O(nm)O(nm)O(nm)
#include<bits/stdc++.h>
#define ll long long
using namespace std;
int n,m,h[2000005];
int dp[2000005];
deque<int>q;
int solve(int k)
{
while(!q.empty()) q.pop_back();
dp[1]=0; q.push_back(1);
//dp[i]=min(dp[j]+(h[j]<=h[i])); j+k[j]-1>=i
for(int i=2;i<=n;i++)
{
while(!q.empty()&&q.front()<i-k) q.pop_front();
dp[i]=dp[q.front()]+(h[q.front()]<=h[i]);
while(!q.empty()&&dp[q.back()]>dp[i]) q.pop_back();
while(!q.empty()&&dp[q.back()]==dp[i]&&h[q.back()]<=h[i]) q.pop_back();
q.push_back(i);
}
return dp[n];
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++) scanf("%d",&h[i]);
cin>>m;
for(int i=1;i<=m;i++)
{
int tmp; scanf("%d",&tmp);
printf("%d\n",solve(tmp));
}
return 0;
}