BZOJ_P3594 [Scoi2014]方伯伯的玉米田(动态规划+二维树状数组优化)

BZOJ传送门

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

Input
第1行包含2个整数n，K，分别表示这排玉米的数目以及最多可进行多少次操作。
第2行包含n个整数，第i个数表示这排玉米，从左到右第i株玉米的高度ai。

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

Sample Input
3 1
2 1 3

Sample Output
3

HINT
1 < N < 10000，1 < K ≤ 500，1 ≤ ai ≤5000

Source
By 佚名提供

Sol:
f[i][j]表示前i个数进行j次操作的最大值

#include<cstdio>
#include<iostream>
using namespace std;
#define N 10005
#define M 5005
#define K 505
#define bit(i) (i&-i)
inline int in(int x=0,char ch=getchar()){while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();return x;}
int d[M+2*K][K],a[N],f[N][K];
int n,m,mm,ans;
inline void add(int x,int y,int z){for(int i=x;i<=mm;i+=bit(i))for(int j=y;j<=m+1;j+=bit(j)) d[i][j]=max(d[i][j],z);}
inline int sum(int x,int y,int res=0){for(int i=x;i;i-=bit(i)) for(int j=y;j;j-=bit(j)) res=max(res,d[i][j]);return res;}
int main(){
    n=in(),m=in();for(int i=1;i<=n;i++) a[i]=in(),mm=max(mm,a[i]);mm+=m;
    for(int i=1;i<=n;i++){
        for(int j=m;~j;--j){
            f[i][j]=sum(a[i]+j,j+1)+1;
            ans=max(ans,f[i][j]);
            add(a[i]+j,j+1,f[i][j]);
        }
    }
    printf("%d\n",ans);
    return 0;
}