洛谷 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+,maxm=1e6+;
int n,h[maxn],f[maxn][],t1[maxm],t0[maxm],ans,mx;
int rd()
{
  int ret=,f=; char ch=getchar();
  while(ch<''||ch>''){if(ch=='-')f=-; ch=getchar();}
  while(ch>=''&&ch<='')ret=(ret<<)+(ret<<)+ch-'',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=; for(;x;x-=(x&-x))ret=max(ret,t0[x]); return ret;}
int query1(int x){int ret=; for(;x<=mx;x+=(x&-x))ret=max(ret,t1[x]); return ret;}
int main()
{
  n=rd();
  for(int i=;i<=n;i++)h[i]=rd()+,mx=max(mx,h[i]);//!
  f[][]=f[][]=; ans=;
  add0(h[],f[][]); add1(h[],f[][]);
  for(int i=;i<=n;i++)
    {
      //      printf("i=%d\n",i);
      f[i][]=query0(h[i]-)+;//!
      f[i][]=query1(h[i]+)+;
      //      printf("f[%d][0]=%d f[%d][1]=%d\n",i,f[i][0],i,f[i][1]);
      add0(h[i],f[i][]); add1(h[i],f[i][]);
      ans=max(ans,max(f[i][],f[i][]));
    }
  printf("%d\n",ans);
  return ;
}