[SCOI2014] 方伯伯的玉米田（最长不降子序列+二维树状数组优化）

[SCOI2014] 方伯伯的玉米田

题目描述

方伯伯在自己的农田边散步，他突然发现田里的一排玉米非常的不美。这排玉米一共有 $N$ 株，它们的高度参差不齐。方伯伯认为单调不下降序列很美，所以他决定先把一些玉米拔高，再把破坏美感的玉米拔除掉，使得剩下的玉米的高度构成一个单调不下降序列。方伯伯可以选择一个区间，把这个区间的玉米全部拔高 $1$ 单位高度，他可以进行最多 $K$ 次这样的操作。拔玉米则可以随意选择一个集合的玉米拔掉。问能最多剩多少株玉米，来构成一排美丽的玉米。

输入格式

第一行包含两个整数 $n,K$，分别表示这排玉米的数目以及最多可进行多少次操作。第二行包含 $n$ 个整数，第 $i$ 个数表示这排玉米，从左到右第 $i$ 株玉米的高度 $a_i$。

输出格式

输出一个整数，最多剩下的玉米数。

样例 #1

样例输入 #1

3 1
2 1 3

样例输出 #1

3

提示

$100\%$ 的数据满足：$2 \le N \lt 10^4 $，$2\le K\le 500$，$1\le a_i\le 5000$。

思路

我们只需要求拔高使得最长的上升子序列的长度即可，如果我们暴力求得话，就是这么写的：

f[i][j] 表示第i株玉米，拔高用了j次的最长不降序列的最大长度。

// for(int i=1;i<=n;i++){
    //     for(int j=m;j>=0;j--){
    //         for(int k=0;k<i;k++){
    //             for(int l=m;l>=0;l--){
    //                 if(w[k]+l<=w[i]+j){
    //                     f[i][j]=max(f[i][j],f[k][l]+1);
    //                 }
    //             }
    //         }
    //     }
    // }
    // cout<<f[n][m];

但这种复杂度很高，$3s$ 都跑不过。

因此我们想，这种上升子序列的问题就是需要快速求出一个二维的前缀最大值，所以引入二维树状数组：

//f[i][j] 表示第i株玉米，拔高用了j次的最长不降序列的最大长度
//其实去除的不用算，只要算最大的上升子序列

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int N = 1e4+10,M = 550;

int tr[M][N];
int n,m;
int w[N];
int mx,res;

int lowbit(int x){
    return x&-x;
}

void add(int x,int y,int c){
    for(int i=x;i<=M;i+=lowbit(i)){
        for(int j=y;j<=N;j+=lowbit(j)){
            tr[i][j]=max(tr[i][j],c);
        }
    }
}

int query(int x,int y){
    int ans=0;
    
    for(int i=x;i;i-=lowbit(i)){
        for(int j=y;j;j-=lowbit(j)){
            ans=max(ans,tr[i][j]);
        }
    }
    return ans;
}

int main(){
    cin>>n>>m;
    
    for(int i=1;i<=n;i++)cin>>w[i],mx=max(mx,w[i]);
    
    //很明显这里是需要快速求出一个二维的前缀最大值，所以引入二位树状数组
    
    // for(int i=1;i<=n;i++){
    //     for(int j=m;j>=0;j--){
    //         for(int k=0;k<i;k++){
    //             for(int l=m;l>=0;l--){
    //                 if(w[k]+l<=w[i]+j){
    //                     f[i][j]=max(f[i][j],f[k][l]+1);
    //                 }
    //             }
    //         }
    //     }
    // }
    // cout<<f[n][m];
    
    for(int i=1;i<=n;i++){
        for(int j=m;j>=0;j--){
            int t=query(j+1,w[i]+j)+1;
            res=max(res,t);
            add(j+1,w[i]+j,t);
        }
    }
    cout<<res;
    
    
    return 0;
}