BZOJ 3594 [SCOI 2014] 方伯伯的玉米田 （DP，树状数组优化）

题目链接：BZOJ 3594

二维树状数组优化DP。
首先写出暴力的动态转移方程：dp[i][j]=max{dp[o][p]}+1.其中满足o<=i && p+a[o]-a[i]<=j(我们把当前状态看做是i由o经过拔高或者是区间去除转移而来).
然后就是优化问题。对于条件p+a[o]-a[i]<=j，将a[i]移项，即为p+a[o]<=j+a[i].我们维护一个二维树状数组。
其中第一维是j+a[i]，第二维是j。但是状态有可能由j=0的转移而来，所以我们将其整体右移一位即可。

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;

const int maxn=10000+10;

int N,K,L=0;

int h[maxn],c[7000][501];

inline int read(){
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}

void input(){
	N=read(); K=read();
	for(int i=1;i<=N;i++){
		h[i]=read(); L=max(L,h[i]);
	}
}

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

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

void update(int x,int y,int v){
	for(int i=x;i<=L+K;i+=lowbit(i))
		for(int j=y;j<=K+1;j+=lowbit(j))
			c[i][j]=max(c[i][j],v);
}

void solve(){
	int ans=0;
	for(int i=1;i<=N;i++){
		for(int j=K;j>=0;j--){
			int t=get(h[i]+j,j+1)+1;
			ans=max(ans,t);
			update(h[i]+j,j+1,t);
		}
	}
	cout<<ans<<endl;
}

int main(){
	input();
	solve();
	return 0;
}