Codeforces 474E Pillars (线段树+DP+二分+离散)

题目链接：http://codeforces.com/problemset/problem/474/E

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

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define read(x,y) scanf("%d%d",&x,&y)
#define ll long long

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
const int  maxn =1e5+5;
const int mod=1e9+7;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}
/*
题目大意:还是给定一个序列要求选择权重最大的满足限制的序列。

还是DP的做法，只不过这次离散要配合二分使用，
把建树的值离散化成下标值，然后有两种做法，
区间查询和单点修改DP值，单点查询和区间修改DP值，
区间修改还要配合懒惰标记还是第一种好一点，
于是一顿操作后，答案值求的出，但还要求答案序列，
最笨的O(n)方法，遍历一遍DP数组不断的往后退找到合适的DP答案。
*/

ll n,d;
ll a[maxn],b[maxn];

int st[maxn<<2],lazy[maxn<<2];
int dp[maxn];
void pushup(lrt){st[rt]=max(st[rt<<1],st[rt<<1|1]);}
void build(lrt)
{
    if(l==r) {st[rt]=0;return;}
    int mid=l+r>>1;
    build(lson),build(rson),pushup(root);
}
void update(lrt,int pos,int d)
{
    if(l==r)
    {
        st[rt]=max(st[rt],d);
        return ;
    }
    int mid=l+r>>1;
    if(pos<=mid) update(lson,pos,d);
    if(mid<pos) update(rson,pos,d);
    pushup(root);
}
int query(lrt,int L,int R)
{
    if(L<=l&&r<=R)  return st[rt];
    int mid=l+r>>1,ans=0;
    if(L<=mid) ans=max(ans,query(lson,L,R));
    if(mid<R) ans=max(ans,query(rson,L,R));
    return ans;
}

void print(int x,int v,ll p)///打印（注意数据范围）
{
    for(int i=x;i>=1;i--)
    {
        if(dp[i]==v&&(a[i]<=p-d||a[i]>=p+d))
        {
            print(i,v-1,a[i]);
            printf("%d ",i);
            break;
        }
    }
}

int main()
{
    scanf("%lld%lld",&n,&d);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        b[i]=a[i];
    }
    sort(b+1,b+n+1);///a[i]+d,a[i]-d;
    ///n=unique(b+1,b+1+n)-b-1;
    build(1,n,1);
    int ans=0;
    for(int i=1;i<=n;i++)
    {
        int id1 = lower_bound(b+1,b+n+1,a[i]+d)-b;
        int id2 = upper_bound(b+1,b+n+1,a[i]-d)-b-1;
        int pos=lower_bound(b+1,b+n+1,a[i])-b;
        if(id2>=1&&b[id2]<=a[i]-d) dp[i]=max(dp[i],query(1,n,1,1,id2));
        if(id1<=n&&b[id1]>=a[i]+d) dp[i]=max(dp[i],query(1,n,1,id1,n));
        dp[i]++;
        update(1,n,1,pos,dp[i]);
        ans=max(ans,dp[i]);
    }
    printf("%d\n",ans);
    int init=0;
    for(int i=1;i<=n;i++)
        if(dp[i]==ans)
        {
            init=i;
            break;
        }
    print(init-1,ans-1,a[init]);printf("%d\n",init);
    return 0;
}