HDU 4521

参考神犇的博客的题目刷下来吧

线段树DP 最长上升子序列的扩展

#include<bits/stdc++.h>
#define N 101000
#define lson root<<1
#define rson root<<1|1
using namespace std;
int dp[N<<2],n,d;
struct mark{
    int val,id;
};
mark a[N];
bool cmp(mark a,mark b){
    if(a.val!=b.val)
        return a.val<b.val;
    return a.id>b.id;
}
void push_up(int root){
    dp[root]=max(dp[lson],dp[rson]);
}
void update(int root,int l,int r,int ql,int qr,int val){
    if(l>qr||r<ql)
        return ;
    if(r<=qr&&l>=ql){
        dp[root]=val;
        return;
    }
    int mid=(l+r)>>1;
    update(lson,l,mid,ql,qr,val);
    update(rson,mid+1,r,ql,qr,val);
    push_up(root);
}
int query(int root,int l,int r,int ql,int qr){
    int res=0;
    if(l>qr||r<ql)
        return 0;
    if(r<=qr&&l>=ql){
        return dp[root];
    }
    int mid=(l+r)>>1;
    res=max(res,query(lson,l,mid,ql,qr));
    res=max(res,query(rson,mid+1,r,ql,qr));
    return res;
}
int main(){
    //freopen("in.txt","r",stdin);
    while(scanf("%d%d",&n,&d)!=EOF){
        memset(dp,0,sizeof(dp));
        for(int i=0;i<n;i++){
            scanf("%d",&a[i].val);
            a[i].id=i+1;
        }
        sort(a,a+n,cmp);
        for(int i=0;i<n;i++){
            int val=0;
            if(a[i].id-d-1>=1) val=query(1,1,n,1,a[i].id-1-d);
            update(1,1,n,a[i].id,a[i].id,val+1);
        }
        printf("%d\n",dp[1]);
    }
}