HihoCoder - 1710 等差子数列(线段树区间最值记录位置，思路)

描述

给定N个整数A1, A2, … AN，小Hi会询问你M个问题。

对于每个问题小Hi给出两个整数L和R(L ≤ R)，请你找出[AL, AL+1, AL+2, …
AR]中最长的等差连续子数列，并输出其长度。

例如[2, 3, 5, 7, 9]中最长的等差连续子数列是[3, 5, 7, 9]长度为4。

输入

第一行包含两个整数N和M。

第二行包含N个整数A1, A2, … AN。

以下M行每行包含两个整数L和R，代表一次询问。

对于30%的数据，1 ≤ N, M ≤ 1000

对于100%的数据，1 ≤ N, M ≤ 100000 0 ≤ Ai ≤ 10000000

输出

依次对于每个询问输出一个整数，代表答案。

样例输入

6 2  
1 2 3 5 7 9  
2 6  
1 4

样例输出

4  
3

思路

这个题要求最长等差连续子数列,对于一个数列，我们可以考虑一下,等差意味着前一项减后一项的值是一个定值，用sub[i]表示第i项与前一项的差，用num[i]表示第1~i个位置的最长等差连续子数列的最大长度(sub[i]相等的个数),很容易发现如果sub[i]==sub[i-1],那么num[i]=num[i-1]+1，如果sub[i]!=sub[i-1]，那么num[i]=2，因为两个数的最大长度肯定是两个.

以样例来进行说明:

下标	1	2	3	4	5	6
a[i]	1	2	3	5	7	9
sub[i]	inf	1	1	2	2	2
num[i]	1	2	3	2	3	4

从上表可以看出，我们好像只需要求出询问区间[l,r]的num[i]的最大值就可以了，试了试样例也觉得能过，但是如果我要求[2,4]这个区间的最长等差连续子数列的值，从上表看出num[i]的值最大是num[3]=3但是很明显，答案应该是2。

我们应该考虑到原因在哪里，首先，区间最大值maxx肯定是要求的，但是，还要顺便记录一下最大值的位置id,比如说区间[2,4]的最大值maxx=3,id=3，我们要先找到id这个位置所表示的最长等差连续子数列，有没有完全被区间[2,4]给包括进去。那么我们只需要判断maxx<=id-r+1就行了，如果小于等于的话，证明这个最大值的区间被完全包括了，此时这个最大值maxx就是答案，但是如果没有被完全包括的话，那么id这个位置所能贡献的最长等差连续子数列的长度只能是id-l+1，也就是这个区间的长度，我们还需要计算出[id+1,r]这个区间的最大值，和id之前的区间最值作比较，取最大值就可以了.

如果[l,r]这个区间的长度小于等于2，也就是l-r+1<=2，当满足这个条件的时候，可以直接输出答案l-r+1

代码

#include <bits/stdc++.h>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define inf 0x3f3f3f3f
const int N=2e5+10;
int n,m;
int a[N],sub[N],num[N];
int MAX[N<<2],id[N<<2];//分别维护最大值和位置
void pushup(int rt)
{
    int ma1=MAX[rt<<1];
    int ma2=MAX[rt<<1|1];
    if(ma1>ma2)
    {
        MAX[rt]=ma1;
        id[rt]=id[rt<<1];
    }
    else
    {
        MAX[rt]=ma2;
        id[rt]=id[rt<<1|1];
    }
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        MAX[rt]=num[l];
        id[rt]=l;
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
struct node
{
    int maxx;//最大值
    int id;//最大值的位置
};
node maxnode(node a,node b)//求出两个结构体的大小
{
    node res;
    if(a.maxx>b.maxx)
    {
        res.maxx=a.maxx;
        res.id=a.id;
    }
    else
    {
        res.maxx=b.maxx;
        res.id=b.id;
    }
    return res;
}
node query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        node res;
        res.maxx=MAX[rt];
        res.id=id[rt];
        return res;
    }
    int m=(l+r)>>1;
    node ans;
    ans.maxx=0;
    if(L<=m) ans=maxnode(ans,query(L,R,lson));
    if(R>m)  ans=maxnode(ans,query(L,R,rson));
    return ans;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&a[i]);
    }
    sub[1]=inf;
    num[1]=1;
    for(int i=2; i<=n; i++)
    {
        sub[i]=a[i]-a[i-1];
        num[i]=(sub[i]==sub[i-1])?num[i-1]+1:2;
    }
    build(1,n,1);//对num[i]建立线段树
    int l,r;
    while(m--)
    {
        scanf("%d%d",&l,&r);
        if(r-l+1<=2)//当区间长度小于等于2
            printf("%d\n",r-l+1);
        else
        {
            node ans=query(l,r,1,n,1);//求出当前区间的最大值和位置
            if(ans.maxx<=ans.id-l+1)//最大值在覆盖范围内
            {
                printf("%d\n",ans.maxx);
            }
            else
            {
                int now_max=ans.id-l+1;//id的左边有多少个
                node res=query(ans.id+1,r,1,n,1);//算出右边最大的
                printf("%d\n",max(now_max,res.maxx));
            }
        }
    }
    return 0;
}