[BZOJbegin][noip2016十连测第八场]降雷皇（dp+线段树）

题目描述

传送门

题解

裸的dp是这样的：
设f(i)表示以h[i]结尾的最长上升子序列的长度，g(i)表示以h[i]结尾的最长上升子序列的方案数。那么$f(i)=max\{f(j)\},j<=i且h_j<h_i；g(i)=\sum\limits_{j=1}^{i-1} g(j)，f(j)+1=f(i)且h_j<h_i$

这个的复杂度是$O(n^2)$的，考虑如何优化。首先f(i)的话可以用二分或者线段树优化做到$O(nlogn)$。而g(i)要麻烦一点，不过也要用到线段树。将所有元素按照f为第一关键字，位置为第二关键字排序，当统计g(i)的时候要保证将f值为f(i)-1且位置在i前面的数都加到线段树里去，在线段树里的位置为h(i)。这样的话每次统计g(i)的时候就只需要求一个前缀和。具体可以用两个指针扫一遍来实现。
也可以用bit来实现。但是由于线段树可以直接打标记清0，所以实现起来比较方便。
时间复杂度$O(nlogn)$。

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define N 100005
#define Mod 123456789

int n,type,Max,top,stack[N],st[N],mf,sg;
struct hp{int id,h,f,g;}a[N];
int sum[N*4],delta[N*4];

void update(int now)
{
    sum[now]=(sum[now<<1]+sum[now<<1|1])%Mod;
}
void pushdown(int now,int l,int r,int mid)
{
    if (delta[now]==0)
    {
        sum[now<<1]=sum[now<<1|1]=delta[now<<1]=delta[now<<1|1]=0;
        delta[now]=-1;
    }
}
void point_change(int now,int l,int r,int x,int v)
{
    int mid=(l+r)>>1;
    if (l==r)
    {
        sum[now]=(sum[now]+v)%Mod;
        return;
    }
    pushdown(now,l,r,mid);
    if (x<=mid) point_change(now<<1,l,mid,x,v);
    else point_change(now<<1|1,mid+1,r,x,v);
    update(now);
}
int query(int now,int l,int r,int lrange,int rrange)
{
    int mid=(l+r)>>1,ans=0;
    if (lrange<=l&&r<=rrange) return sum[now];
    pushdown(now,l,r,mid);
    if (lrange<=mid) ans=(ans+query(now<<1,l,mid,lrange,rrange))%Mod;
    if (mid+1<=rrange) ans=(ans+query(now<<1|1,mid+1,r,lrange,rrange))%Mod;
    return ans;
}

int find(int x)
{
    int l=1,r=top,mid,ans;
    while (l<=r)
    {
        int mid=(l+r)>>1;
        if (stack[mid]>=x) ans=mid,r=mid-1;
        else l=mid+1;
    }
    return ans;
}
int cmp(hp a,hp b)
{
    return a.f<b.f||(a.f==b.f&&a.id<b.id);
}
int main()
{
    scanf("%d%d",&n,&type);
    for (int i=1;i<=n;++i)
    {
        scanf("%d",&a[i].h);a[i].id=i;
        Max=max(Max,a[i].h);
    }

    for (int i=1;i<=n;++i)
    {
        if (a[i].h>stack[top])
        {
            stack[++top]=a[i].h;
            a[i].f=top;mf=max(mf,a[i].f);
        }
        else
        {
            int loc=find(a[i].h);
            stack[loc]=a[i].h;
            a[i].f=loc;mf=max(mf,a[i].f);
        }
    }
    if (type!=1)
    {
        printf("%d\n",mf);
        return 0;
    }
    sort(a+1,a+n+1,cmp);
    for (int i=1;i<=n;++i)
        if (a[i].f!=a[i-1].f) st[a[i].f]=i;
    st[mf+1]=n+1;
    for (int i=st[1];i<=n&&i!=st[2];++i) a[i].g=1;
    memset(delta,-1,sizeof(delta));
    for (int i=1;i<mf;++i)
    {
        int l=st[i],r=st[i+1];
        if (!l||!r) continue;
        while (r<st[i+2])
        {
            while (l<st[i+1]&&a[l].id<a[r].id)
                point_change(1,1,Max,a[l].h,a[l].g),l++;
            if (a[r].h>=1) a[r].g=query(1,1,Max,1,a[r].h-1);
            r++;
        }
        sum[1]=0; delta[1]=0;
    }
    for (int i=1;i<=n;++i)
        if (a[i].f==mf) sg=(sg+a[i].g)%Mod;
    printf("%d\n%d\n",mf,sg);
}

总结

①注意边界条件。
②注意线段树的区间操作左区间和右区间判断大小。