lnsyoj106猴子

▼魔方 西西 提交于 2020-02-16 02:50:55

这道题的算法是一个动态规划加单调队列优化时间复杂度,未优化的时间复杂度是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;
}
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!