Codeforces Round #271 (Div. 2) E Pillars（dp+线段树优化）

题目要求abs(d[i]-d[j])>=d，也就是说只要a[j]<=a[i]-d或者a[i]+d<=a[j] ，也就是只要求1~a[i]-d即a[i]+d~max里最大的dp值，这就是一个线段树查询问题。建议先看一下本博客里面LIS线段树优化的文章。

下面是AC代码：

  

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long
using namespace std;
int dp[100005],pre[100005];
LL s[100005],a[100005];
int b[100005];
struct pi{
    int le;
    int ri;
    int ma;
}pp[400005];
void init(int tot,int l,int r){
    pp[tot].ma=-1;
    pp[tot].le=l;
    pp[tot].ri=r;
    if(l==r) return ;
    init(2*tot,l,(l+r)/2);
    init(2*tot+1,(l+r)/2+1,r);
}
void merg(int tot,int p,int n){
    if(pp[tot].le==pp[tot].ri){
        if(pp[tot].ma==-1) pp[tot].ma=n;
        else{
            if(dp[pp[tot].ma]<dp[n]){
                pp[tot].ma=n;
            }
        }
        return ;
    }
    int mid=(pp[tot].le+pp[tot].ri)/2;
    if(p<=mid) merg(2*tot,p,n);
    else merg(2*tot+1,p,n);
    if(pp[2*tot].ma==-1) pp[tot].ma=pp[2*tot+1].ma;
    if(pp[2*tot+1].ma==-1) pp[tot].ma=pp[2*tot].ma;
    if(dp[pp[2*tot].ma]<dp[pp[2*tot+1].ma]) pp[tot].ma=pp[2*tot+1].ma;
    else pp[tot].ma=pp[2*tot].ma;
}
int query(int tot,int l,int r){
    int x,y;
    if(l<=pp[tot].le&&r>=pp[tot].ri) return pp[tot].ma;
    int mid;
    mid=(pp[tot].le+pp[tot].ri)/2;
    x=-1;
    y=-1;
    if(l<=mid) x=query(2*tot,l,r);
    if(r>mid) y=query(2*tot+1,l,r);
    if(x==-1) return y;
    if(y==-1) return x;
    if(dp[x]<dp[y]) return y;
    return x;
}
int get(int n,LL k){
    int le,ri,mid;
    le=1;
    ri=n;
    while(le<=ri){
        mid=(le+ri)/2;
        if(s[mid]<k) le=mid+1;
        else ri=mid-1;
    }
    return le;
}
int main()
{
    int i,j,n,m,p,k,d,x,y;
    LL a1,b1;
    cin>>n>>d;
    for(i=1;i<=n;i++){
        cin>>a[i];
        s[i]=a[i];
    }
    sort(s+1,s+1+n);
    dp[0]=0;
    init(1,1,n);
    memset(pre,-1,sizeof(pre));
    for(i=1;i<=n;i++){
        if(i==1){
            dp[1]=1;
            p=get(n,a[i]);
            merg(1,p,1);
        }
        else{
            x=-1;
            y=-1;
            a1=a[i]-d;
            if(a1>0){
                p=get(n,a1);
                if(p>n||s[p]!=a1) p--;
            }
            else p=0;
            if(p>0)
            x=query(1,1,p);
            b1=a[i]+d;
            p=get(n,b1);
            if(p<=n) y=query(1,p,n);
            if(x==-1&&y==-1) dp[i]=1;
            else{
                if(x==-1){
                    dp[i]=dp[y]+1;
                    pre[i]=y;
                }
                else if(y==-1){
                    dp[i]=dp[x]+1;
                    pre[i]=x;
                }
                else{
                    if(dp[x]<dp[y]){
                        dp[i]=dp[y]+1;
                        pre[i]=y;
                    }
                    else {
                        dp[i]=dp[x]+1;
                        pre[i]=x;
                    }
                }
            }
            p=get(n,a[i]);
            merg(1,p,i);
        }
    }
    p=0;
    for(i=1;i<=n;i++){
        if(dp[i]>p) p=dp[i];
    }
    int tot;
    printf("%d\n",p);
    tot=0;
    for(i=1;i<=n;i++){
        if(dp[i]==p){
            b[tot++]=i;
            i=pre[i];
            while(i!=-1){
                b[tot++]=i;
                i=pre[i];
            }
            break;
        }
    }
    for(i=tot-1;i>=0;i--) printf(" %d",b[i]);
    printf("\n");
}