lnsyoj106猴子

这道题的算法是一个动态规划加单调队列优化时间复杂度,未优化的时间复杂度是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;
}
发布了20 篇原创文章 · 获赞 0 · 访问量 266

猜你喜欢

转载自blog.csdn.net/weixin_45885433/article/details/104307855
106