洛谷 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\n",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\n",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\n",ans); return 0; }