【BZOJ 4491】我也不知道题目名字是什么【差分+线段树维护】

【前言】又来切水题了） -__-。。。

Description

给定一个序列A[i]，每次询问l,r，求[l,r]内最长子串，使得该子串为不上升子串或不下降子串(n < 50000)

Input

第一行n，表示A数组有多少元素
接下来一行为n个整数A[i]
接下来一个整数Q，表示询问数量
接下来Q行，每行2个整数l，r

Output

对于每个询问，求[l,r]内最长子串，使得该子串为不上升子串或不下降子串

题解

　　考虑将原数列差分，那么最长不下降字串意味着最长的大于等于0的字串，最长不上升也一样。这样就成了一道 傻逼 经典题。

　　开两个线段树感觉做起来更方便啊，每个都保存区间最长、从左开始最长、从右开始最长。然后向上统计信息时做几个简单的判断就行了，具体还是见我代码吧。

　　最后要查询的区间分别是[l+1,r]（不下降），[l,r-1]（不上升），我在此WA了3次。因为我们可以将查分后的正负看做函数的一个增减线段，而每条线段是对应两个端点，所以要查询那样的区间，同时，因为对应两个端点，所以ans还要加上1。

代码

#include<cstdio>
#include<cstring>

#define N 50010
using namespace std;
int fl[N << 2][2],fr[N << 2][2],sum[N << 2][2];
int a[N],b[N];
int n,m,ll,rr;

#define lson o << 1
#define rson o << 1 | 1
inline int max(int x,int y){return x > y ? x : y;}
inline int min(int x,int y){return x < y ? x : y;}
inline int swap(int &x,int &y){int t = x;x = y;y = t;}

void pushup(int o,int l,int r,int k)
{
    sum[o][k] = max(max(sum[lson][k],sum[rson][k]),fr[lson][k]+fl[rson][k]);
    int mid = (l+r)>>1;
    fl[o][k] = fl[lson][k] + (fl[lson][k] == mid-l+1 ? fl[rson][k] : 0);
    fr[o][k] = fr[rson][k] + (fr[rson][k] == r-mid ? fr[lson][k] : 0);
}

void build(int o,int l,int r,int k)
{
    if(l == r){fl[o][k] = fr[o][k] = sum[o][k] = (b[l] >= 0 ? 1 : 0);return;}
    int mid = (l+r)>>1;
    build(lson,l,mid,k); build(rson,mid+1,r,k);
    pushup(o,l,r,k);
}

int query(int o,int l,int r,int ll,int rr,int k)
{
    if(ll <= l && rr >= r) return sum[o][k];
    int mid = (l+r)>>1;
    int ans = 0,l1 = 0,r1 = 0;
    if(ll <= mid) {
        ans = max(ans,query(lson,l,mid,ll,rr,k));
        if(rr > mid) l1 = min(mid-ll+1,fr[lson][k]);
    }
    if(rr > mid) {
        ans = max(ans,query(rson,mid+1,r,ll,rr,k));
        if(ll <= mid) r1 = min(rr-mid,fl[rson][k]);
    }
    ans = max(ans,l1+r1);
    return ans;
}

int main()
{
    scanf("%d",&n);
    for(int i = 1;i <= n;i++) scanf("%d",&a[i]);
    for(int i = 1;i <= n;i++) b[i] = a[i] - a[i-1];
    build(1,1,n,0);
    for(int i = 1;i <= n;i++) b[i] = a[i] - a[i+1];
    build(1,1,n,1);
    scanf("%d",&m);
    while(m--) {
        scanf("%d%d",&ll,&rr);
        if(ll == rr) {puts("1");continue;}
        if(ll > rr) swap(ll,rr);
        printf("%d\n",max(query(1,1,n,ll+1,rr,0),query(1,1,n,ll,rr-1,1))+1);//这里很重要!!!
    }
    return 0;
}