bzoj3594 [Scoi2014]方伯伯的玉米田

题目链接

二维树状数组优化DP

DP状态很容易想到：dp[i][j]表示到第i颗玉米，用了j次提升，最多保留多少：

转移：

　　　　dp[i][j]=dp[k][j]+1(k<=i&&a[k]<=a[i)

　　　　dp[i][j]=dp[k][j+a[i]-a[k]]+1 (j<=i&&a[k]<=a[i])

这两种转移都可以用二维树状数组优化：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<ctime>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define lowbit(x) x&-x
using namespace std;
int c[6060][555],hh[10010][555],a[10010],n,m,ans,Max; 
int getint()
{
    int ret=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')f=0;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')ret*=10,ret+=ch-'0',ch=getchar();
    return f==1?ret:-ret;
}
void add(int a,int b,int x)
{
    for(int i=a;i<=Max+m;i+=lowbit(i))
        for(int j=b;j<=m+1;j+=lowbit(j))
            c[i][j]=max(c[i][j],x);
}
int query(int a,int b)
{
    int ret=0;
    for(int i=a;i;i-=lowbit(i))
        for(int j=b;j;j-=lowbit(j))
            ret=max(c[i][j],ret);
    return ret;
}
int main()
{
    n=getint(),m=getint();
    for(int i=1;i<=n;i++)
    {
        a[i]=getint();
        Max=max(Max,a[i]);
    }
    for(int i=1;i<=n;i++)
        for(int j=m;j>=0;j--)
        {
            hh[i][j]=query(a[i]+j,j+1)+1;
            ans=max(ans,hh[i][j]);
            add(a[i]+j,j+1,hh[i][j]);
        }
    printf("%d
",ans);
    return 0;
}