这道题的算法是一个动态规划加单调队列优化时间复杂度,未优化的时间复杂度是O(nnm),优化后是O(nm)。
构造一个单减队列,队头是起始点,新入队的元素是要跳跃到的树。
代码如下
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
using namespace std;
int val[5002],s[5002],q[5002];//q里存放的是树的数
int dp[5002][5002];//第一维保存猴子跳跃的次数,第二维保存第几棵树
int main()
{
int n,m,d;
scanf("%d%d%d",&n,&d,&m);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&val[i],&s[i]);
}
dp[0][1]=val[1];
int head,tail;
int ans=-1000;
for(int i=1;i<=m;i++)//两层循环都是枚举树!!!!但要维护m次跳跃次数
{
if(s[i+1]-s[i]>d)//因为数据很迷惑 所以需要特判一下
{
for(int j=1;j<=i;j++)
ans=max(ans,dp[i-1][j]);
printf("%d\n",ans);
return 0;
}
head=tail=1;
q[tail]=i;//第i棵树入队
for(int j=i+1;j<=n;j++)
{
while(head<=tail&&(s[j]-s[q[head]])>d)//队头是现在猴子的所在位置
{
head++;
}
dp[i][j]=max(dp[i-1][q[head]]+val[j],dp[i][j]);//转移方程
while(head<=tail&&dp[i-1][q[tail]]<dp[i-1][j])
{
tail--;//维护单减队列
}
q[++tail]=j;//j入队
}
}
for(int i=1;i<=n;i++)
{
ans=max(ans,dp[m][i]);//跳跃m次终点为i
}
printf("%d",ans);
return 0;
}
来源:CSDN
作者:Christine_xb
链接:https://blog.csdn.net/weixin_45885433/article/details/104307855