[NOIP2013][vijos1845]花匠（dp+bit）

题目描述

传送门

题解

由题意可知要求留下的花的高度成锯齿形。
设f(i,0)表示以i结尾的花，是上升趋势 留下的最多的花，f(i,1)表示以i结尾的花，是下降趋势 留下最多的花。
那么f(i,0)=max{f(j,1)},1<=j < <script type="math/tex" id="MathJax-Element-179"><</script>i且h(j) < <script type="math/tex" id="MathJax-Element-180"><</script>h(i),f(i,1)=max{f(j,0)},1<=j < <script type="math/tex" id="MathJax-Element-181"><</script>i且h(j)>h(i).
最暴力的转移是$O(n)$的，用树状数组优化一下可以做到$O(logn)$转移，总复杂度$O(nlogn)$。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define N 1000005

int n,Max,h[N];
int f[N][2],C0[N],C1[N];

void change0(int loc,int val)
{
    for (int i=loc;i<=Max;i+=i&(-i))
        C0[i]=max(C0[i],val);
}
void change1(int loc,int val)
{
    for (int i=loc;i<=Max;i+=i&(-i))
        C1[i]=max(C1[i],val);
}
int query0(int loc)
{
    int ans=0;
    for (int i=loc;i>=1;i-=i&(-i))
        ans=max(ans,C0[i]);
    return ans;
}
int query1(int loc)
{
    int ans=0;
    for (int i=loc;i>=1;i-=i&(-i))
        ans=max(ans,C1[i]);
    return ans;
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;++i) scanf("%d",&h[i]),h[i]++,Max=max(Max,h[i]);
    f[1][0]=1;f[1][1]=1;
    change0(Max-h[1]+1,f[1][0]);
    change1(h[1],f[1][1]);
    for (int i=2;i<=n;++i)
    {
        f[i][0]=query1(h[i]-1);
        f[i][0]++;
        change0(Max-h[i]+1,f[i][0]);
        f[i][1]=query0(Max-h[i]+1-1);
        f[i][1]++;
        change1(h[i],f[i][1]);
    }
    printf("%d\n",max(f[n][0],f[n][1]));
}