洛谷 P1970 花匠 —— DP

题目：https://www.luogu.org/problemnew/show/P1970

普通的DP，f[i][0/1] 表示 i 处处于较小或较大的长度；

注意：1.树状数组向后 query 时不能有0，所以读入的 h[i] 都+1；

2.可能有相同的高度！所以 query 时注意 +1,-1。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int const maxn=1e5+5,maxm=1e6+5;
int n,h[maxn],f[maxn][3],t1[maxm],t0[maxm],ans,mx;
int rd()
{
  int ret=0,f=1; char ch=getchar();
  while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
  while(ch>='0'&&ch<='9')ret=(ret<<3)+(ret<<1)+ch-'0',ch=getchar();
  return ret*f;
}
void add0(int x,int val){for(;x<=mx;x+=(x&-x))t0[x]=max(t0[x],val);}
void add1(int x,int val){for(;x;x-=(x&-x))t1[x]=max(t1[x],val);}
int query0(int x){int ret=0; for(;x;x-=(x&-x))ret=max(ret,t0[x]); return ret;}
int query1(int x){int ret=0; for(;x<=mx;x+=(x&-x))ret=max(ret,t1[x]); return ret;}
int main()
{
  n=rd(); 
  for(int i=1;i<=n;i++)h[i]=rd()+1,mx=max(mx,h[i]);//!
  f[1][0]=f[1][1]=1; ans=1;
  add0(h[1],f[1][0]); add1(h[1],f[1][1]);
  for(int i=2;i<=n;i++)
    {
      //      printf("i=%d
",i);
      f[i][1]=query0(h[i]-1)+1;//!
      f[i][0]=query1(h[i]+1)+1;
      //      printf("f[%d][0]=%d f[%d][1]=%d
",i,f[i][0],i,f[i][1]);
      add0(h[i],f[i][0]); add1(h[i],f[i][1]);
      ans=max(ans,max(f[i][0],f[i][1]));
    }
  printf("%d
",ans);
  return 0;
}