DongDong跳一跳 DP+权值线段树优化 牛客

链接：https://ac.nowcoder.com/acm/contest/904/C

题面:
DongDong有一只超可爱的英短喜欢跳一跳，但此跳一跳非彼跳一跳
有n根柱子，每根柱子都有一个高度和柱子上面鱼干的数量，英短开始的时候可以选择站在任意一根柱子上，每次跳跃不限长度而且只能从左向右跳跃，但只能跳到高度与当前所站高度差绝对值小于等于m的柱子上，英短可贪心了，想吃最多的鱼干，请你设计一个程序能让英短吃到最多的鱼干（最终不一定要落在第n根柱子上）。
思路: 在每个位置都可以由前面某个范围高度转移得到，所以用权值线段树O(logn)的时间就可以快速找到前面最大的值

#include<bits/stdc++.h>
#define lson l,mid,root<<1
#define rson mid+1,r,root<<1|1
using namespace std;
typedef long long ll;
const int N=2e5+5;
const int M=1e6+1000;
int a[N],b[N];
ll ans;
ll dp[N];
ll val[M<<2];
void pushup(int root)
{
    val[root]=max(val[root<<1],val[root<<1|1]);
    return ;
}

void build(int l,int r,int root)
{
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(root);
}

void update(int l,int r,int root,ll x,int pos)
{
    if(l==r)
    {
        val[root]=max(val[root],x);
        return ;
    }
    int mid=(l+r)>>1;
    if(pos<=mid)
        update(lson,x,pos);
    else
        update(rson,x,pos);
    pushup(root);
    return ;
}

ll query(int L,int R,int l,int r,int root)
{
    if(L<=l&&r<=R)
        return val[root];
    int mid=(l+r)>>1;
    ll res=0;
    if(L<=mid)
        res=max(res,query(L,R,lson));
    if(R>mid)
        res=max(res,query(L,R,rson));
    return res;
}

int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    int maxx=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&a[i],&b[i]);
        maxx=max(maxx,a[i]);
    }
    maxx+=m;
    build(1,maxx,1);
    for(int i=1;i<=n;i++)
    {
        dp[i]=query(max(1,a[i]-m),a[i]+m,1,maxx,1)+b[i];
        update(1,maxx,1,dp[i],a[i]);
        ans=max(dp[i],ans);
    }
    printf("%lld\n",ans);
    return 0;
}