NOIP2017 PJ 跳房子 —— 单调队列优化DP

题目描述###

跳房子,也叫跳飞机,是一种世界性的儿童游戏,也是中国民间传统的体育游戏之一。跳房子的游戏规则如下:
在地面上确定一个起点,然后在起点右侧画n个格子,这些格子都在同一条直线上。每个格子内有一个数字(整数),表示到达这个格子能得到的分数。玩家第一次从起点开始向右跳,跳到起点右侧的一个格子内。第二次再从当前位置继续向右跳,依此类推。规则规定:玩家每次都必须跳到当前位置右侧的一个格子内。玩家可以在任意时刻结束游戏,获得的分数为曾经到达过的格子中的数字之和。
现在小R研发了一款弹跳机器人来参加这个游戏。但是这个机器人有一个非常严重的缺陷,它每次向右弹跳的距离只能为固定的d。小R希望改进他的机器人,如果他花g个金币改进他的机器人,那么他的机器人灵活性就能增加g,但是需要注意的是,每次弹跳的距离至少为1。具体而言,当g<d时,他的机器人每次可以选择向右弹跳的距离为d-g,d-g+1,d-g+2,…,d+g-2,d+g-1,d+g;否则(当g≥d时),他的机器人每次可以选择向右弹跳的距离为1,2,3,…,d+g-2,d+g-1,d+g。
现在小R希望获得至少k分,请问他至少要花多少金币来改造他的机器人。

输入输出格式###

输入格式:
第一行三个正整数 n , d , k ,分别表示格子的数目,改进前机器人弹跳的固定距离,以及希望至少获得的分数。相邻两个数 之间用一个空格隔开。
接下来 n 行,每行两个正整数 xi, si ,分别表示起点到第 i 个格子的距离以及第 i 个格子的分数。两个数之间用一个空格隔开。保证 xi 按递增顺序输入。

输出格式:
共一行,一个整数,表示至少要花多少金币来改造他的机器人。若无论如何他都无法获得至少 k 分,输出 -1 。

输入输出样例###

输入样例#1:
7 4 10
2 6
5 -3
10 3
11 -3
13 1
17 6
20 2
输出样例#1:
2
样例1说明:
2个金币改进后, 小 R 的机器人依次选择的向右弹跳的距离分别为2,3,5,3,4,3, 先后到达的位置分别为 2,5,10,13,17,20, 对应1,2,3,5,6,7 这6 个格子。这些格子中的数字之和 15 即为小 R 获得的分数。

样例 2输入:
7 4 20
2 6
5 -3
10 3
11 -3
13 1
17 6
20 2
样例2输出:
-1
样例2说明:
由于样例中 7 个格子组合的最大可能数字之和只有 18 ,无论如何都无法获得20分。

解题思路###

通过对问题进行分析,发现“花的钱越多跳到范围越广,更容易满足条件”,显然答案有单调性,所以二分答案。
二分花多少钱,检验能否拿到k分,考虑怎么Check?
答案的check可以DP,因为从左往右跳跃的过程是线性的。
看出来这应该是一个二分答案验证的套路,我们二分花费金币的数量,通过这个值确定跳跃的长度范围,可以用这样的DP来计算到每个位置为终点的最大路径和。
f[i]表示跳前i个格子,且停在第i个格子最大分数;
num[i]表示第i个格子的分数。
于是:$ f[i]=max{ f[j] | 从j可以跳到i} + num[i] $

这样就可以判断答案的合法性了,有这么简单,它不会这么简单,算一下时间复杂度:\(Θ(logN * N^2)\)
实现代码:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define N 500005
#define ll long long

int n,d,sco;
ll s[N],f[N],dis[N];
 
int pd(int g) {
	int lpos=d-g;  //跳的最短距离 
    int rpos=d+g;  //跳的最长距离 
    if(lpos<=0) lpos = 1;
    memset(f,-127,sizeof(f));//设为很小的负数
    f[0]=0;
	for(int i=1;i<=n;++i) {
		for(int j=i-1;j>=0;j--) {  
			if(dis[i]-dis[j]<lpos) continue;
			if(dis[i]-dis[j]>rpos) break;
			f[i]=max(f[i],f[j]+s[i]);
			if(f[i]>=sco) return 1;
		}
	}
	return 0;
}
 
ll Bsearch() {
	ll l=0,r=dis[n]+1,mid,ans=-1; // 不一定非得一个一个跳,那样可能会到不了sco 
	while(l<r) {
		mid=(l+r)>>1;
		if(pd(mid)) r=mid;
		else l=mid+1;
	}
	return r;
}
 
int main()
{
	cin>>n>>d>>sco;
	ll sum=0;
	for(ll i=1;i<=n;++i) { 
		cin>>dis[i]>>s[i];
		if(s[i]>0) sum+=s[i];
	}
	if(sum>=sco) printf("%lld\n",Bsearch());
	else printf("-1\n");
	return 0;
}

因此,要考虑优化Check中的DP……
发现转移式中的f[j]是随i的右移而右移的,即:如果每次都枚举,会重复枚举很多个f[j],如果可以把这些f[j]保存下来就好了。
由于跳跃范围是定长区间,我们想到了滑动窗口,显然可以用单调队列来维护定长区间最大值,从而完成的O(1)转移。
单调队列是一个双向队列(deque),即两头都可以push或者pop等。
怎么优化呢?假设此时队列里面已经有了f[i−1]所枚举过的f[j],要得到f[i]所对应的max(f[j]),只需要把新加入的加入,不符合条件的扔出去即可。
发现,对于格子j,如果j到i的距离已经大于了机器人能跳的最大距离,它就可以扔掉了,因为i是一直往右走的。
对于格子j,如果j到i的距离大于等于机器人能跳的最小距离,显然f[j]是可以入队的,但是在把f[j]放进去之前要和队列最后面的元素f[k]比较,如果f[k]<f[j],f[k]就可以直接pop掉了。
于是,这个队列中的元素一定是下降的(注意不是优先队列,优先队列是一个堆),使用时直接取front即可。
事实上,在单调队列中存下标即可,实现时可以直接用STL中的deque(双向队列)。
实现代码

#include<bits/stdc++.h>
using namespace std;

#define MAXN 500050
int pos[MAXN+5],num[MAXN+5];
int N,D,K;
int f[MAXN+5];

bool dp_check(int g)
{
    deque<int> Q;
    memset(f,0,sizeof f);
    int Lg=max(1,D-g),Rg=D+g;
    int Now=0;        //Now为当前新加入Q的格子
    for(int i=1;i<=N;i++)
    {
        while(pos[Now]+Lg<=pos[i])
        {
            while(!Q.empty()&&f[Q.back()]<=f[Now]) //!Q.empty要写在前面,否则Q.back()报错 
                Q.pop_back();
            Q.push_back(Now++);                    //不一定只放一个
        }
        while(!Q.empty()&&pos[Q.front()]+Rg<pos[i])//把队列前面不需要的pop掉
              Q.pop_front();
        if(!Q.empty()) f[i]=f[Q.front()]+num[i];   //直接取出最前面的最大值进行转移 
        else f[i]=-0x3f3f3f3f;                //否则队列为空,表示到不了这个格子
        if(f[i]>=K) return 1; //往右递推时随时都有可能>=K,而不是只在循环完后再比较f[N]与K的大小 
    }
    return 0;
}

int main()
{
    cin>>N>>D>>K;
    long long sum=0;
    for(int i=1;i<=N;i++)
    {
        cin>>pos[i]>>num[i];
        if(num[i]>0) sum+=num[i];
    }
    if(sum<K)
    {
        printf("-1\n");
        return 0;
    }
    int left=0,right=pos[N];//注意答案的最大值是 pos[N],而不是N!! 
    while(left<right)
    {
        int mid=(left+right)>>1;
        if(dp_check(mid)) right=mid;
        else left=mid+1;//mid不符合条件,left右移是 mid+1
    }
    printf("%d\n",right);
    return 0;
}
posted @ 2019-10-24 11:52  _tham  阅读(355)  评论(0编辑  收藏  举报