[SCOI2014] 方伯伯的玉米田
题目描述
方伯伯在自己的农田边散步,他突然发现田里的一排玉米非常的不美。这排玉米一共有 N N N 株,它们的高度参差不齐。方伯伯认为单调不下降序列很美,所以他决定先把一些玉米拔高,再把破坏美感的玉米拔除掉,使得剩下的玉米的高度构成一个单调不下降序列。方伯伯可以选择一个区间,把这个区间的玉米全部拔高 1 1 1 单位高度,他可以进行最多 K K K 次这样的操作。拔玉米则可以随意选择一个集合的玉米拔掉。问能最多剩多少株玉米,来构成一排美丽的玉米。
输入格式
第一行包含两个整数 n , K n, K n,K,分别表示这排玉米的数目以及最多可进行多少次操作。第二行包含 n n n 个整数,第 i i i 个数表示这排玉米,从左到右第 i i i 株玉米的高度 a i a_i ai。
输出格式
输出一个整数,最多剩下的玉米数。
样例 #1
样例输入 #1
3 1
2 1 3
样例输出 #1
3
提示
100 % 100\% 100% 的数据满足:$2 \le N \lt 10^4 $, 2 ≤ K ≤ 500 2 \le K \le 500 2≤K≤500, 1 ≤ a i ≤ 5000 1 \leq a_i \leq 5000 1≤ai≤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];
但这种复杂度很高, 3 s 3s 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;
}