[BZOJ3594][DP][树状数组]SCOI2014：方伯伯的玉米田

BZOJ3594

先分析一下题：要求执行k次区间加1后这个区间最长的LIS长度

那么首先考虑区间加1这个操作：很明显这个操作的右端点一定可以为n，如果最优决策中有r不为n的情况，把r后面的数都加1，也对答案没有影响，这个很明显

所以就可以考虑dp：设$f[i][j]$表示dp到第i个，已经执行了j次区间加1操作的ans，转移方程就是$f[i][j]=max(f[a][b])+1,(a<i,b\le j,val[a]+b\le val[i]+j)$，显然用树状数组维护

Code：

#include<bits/stdc++.h>
#define lb(x) x&-x
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int K=505,N=10005;
int dp[N][K],a[N];
int n,k,ans,maxn;
namespace Bit{
	int bit[6005][K];
	inline void add(int x,int y,int v){
		for(int i=x;i<=maxn+k;i+=lb(i))
			for(int j=y;j<=k+1;j+=lb(j)) bit[i][j]=max(bit[i][j],v);
	}
	inline int ask(int x,int y){
		int res=0;
		for(int i=x;i;i-=lb(i))
			for(int j=y;j;j-=lb(j)) res=max(res,bit[i][j]);
		return res;
	}
}
int main(){
	n=read();k=read();
	int ans=0;
	for(int i=1;i<=n;i++) a[i]=read(),maxn=max(maxn,a[i]);
	for(int i=1;i<=n;i++)
		for(int j=k;j>=0;j--){
			dp[i][j]=Bit::ask(a[i]+j,j+1)+1;
			ans=max(ans,dp[i][j]);
			Bit::add(a[i]+j,j+1,dp[i][j]);
		}
	cout<<ans<<endl;
	return 0;
}