等差子数列 HihoCoder - 1710

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。

Input
第一行包含两个整数N和M。
第二行包含N个整数A1, A2, … AN。
以下M行每行包含两个整数L和R，代表一次询问。
对于30%的数据，1 ≤ N, M ≤ 1000
对于100%的数据，1 ≤ N, M ≤ 100000 0 ≤ Ai ≤ 10000000

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

先写了一发奇奇怪怪的线段树。。。虽然知道大概哪里错了，但是没改出来。。。
最后用的是RMQ
首先举个栗子
1 2 3 5 7 9 8 7 6 5
我们可以看出来1 2 3 是个等差数列，3 5 7 9 是个等差数列，9 8 7 6 5是个等差数列，那么我们可以把这个数组分为三段

下面是几个数组的变量解释;
sum[i] : 第i个数字所在的等差数列的等差值
L[i] ：第i个数字所在的等差数列的最右端的开始的下标
R[i] ：第i个数字所在的等差数列最左边的结束的下标
那么刚刚上面的那个例子的sum，L，R的值就是下面的表

	1	2	3	5	7	9	8	7	6	5
sum	1	1	2	2	2	-1	-1	-1	-1	-1
L	1	1	3	3	3	6	6	6	6	6
R	3	3	6	6	6	10	10	10	10	10

那么我们把每次查询的【l,r】的结果分为三个部分的最大值，R[l]-l+1,r-L[r]+1,R[l]+1~L[r]-1
分别代表了从l到l所在的段的结束处的元素个数，从r所在段的开始处到r处的元素个数，和这中间的那一段的sum的值
那么如果我们想要计算【2,9】之间的最长连续等差序列，也就是2 3 5 7 9 8 7 6 之间的最长连续子序列
那么对于这次查询，可以把这个查询根据等差数列分的段分为3段2 3 | 5 7 | 9 8 7 6这三段
2 3这一段一共有2个数
5 7这一段的最大的sum值是3
9 8 7 6这一段一共有4个数
所以这一段的最长连续子序列的个数就是4
对于中间段R[l]+1到L[r]-1这个区间里找sum的最大值用的是RMQ查找。。。

特别的

如果l和r在同一个等差数列的段中，那么答案就是r-l+1；

* 对于查询中间段R[l]+1到L[r]-1这个区间里找sum的最大值这个步骤，要用到RMQ我以前一直写线段树，第一次写RMQ，还是记录一下：

首先预处理一下dp数组（听大佬讲，RMQ用的思想是DP）：
dp[i][j]:第i位数字到它后面的2^j个位置的最大值（i后面的一个位置是它本身）
首先预处理dp[i][0]=sum[i];
下面的我们来举个栗子
现在有4个数字 1 2 3 4 5
我们要算dp[1][2],也就是1后面的包括它本身的四个数字
那么我们可以把1 2 3 4 分为1 2 一段 3 4 一段
1 2 是1后面包括他本身的两个数字 他们的最大值也就是 dp[i][2-1];
3 4 shi 3后面包括他本身的两个数字，3和1之间隔了2¹个数字，也就是dp[i+(1<<(j-1))][j-1];
所以dp[i][j]=max (dp[i][j-1] , dp[i+(1<<(j-1))][j-1]);
然后查询的时候，令k为满足2^k<=r-l+1的最大整数，则以l开头，以r结尾的 两个 长度为2^k的区间合起来覆盖了查询区间【l，r】，取最大值就好了

#include<algorithm>
#include<stdio.h>
#include<string.h>
using namespace std;
int n,m,ans;
int A[100010];
int L[100010],R[100010],sum[100010],dp[100010][25];
void init_RMQ()
{
    for(int i=1; i<=n; i++)dp[i][0]=sum[i];
    for(int j=1; (1<<j)<=n; j++)
    {
        for(int i=1; i+(1<<j)-1<=n; i++)
        {
            dp[i][j]=max(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
        }
    }
}
int RMQ(int l,int r)
{
    if(r<l)
        return -1;
    int k=0;
    while(1<<(k+1)<=r-l+1) k++;
    return max(dp[l][k],dp[r-(1<<k)+1][k]);
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1; i<=n; i++)
            scanf("%d",&A[i]);
        int i=1,j,ca,cut,q;
        while(i<=n)
        {
            q=i;
            ca=A[i+1]-A[i];
            j=i;
            cut=1;
            while(j<=n&&A[j+1]-A[j]==ca) j++,cut++;
            for(; i<j; i++)
            {
                L[i]=q,R[i]=j,sum[i]=cut;
            }
        }
        for(int i=1; i<=n; i++)
            printf("%d ",L[i]);
        printf("\n");
        for(int i=1; i<=n; i++)
            printf("%d ",R[i]);
        init_RMQ();
        int l,r;
        while(m--)
        {
            scanf("%d%d",&l,&r);
            ans=min(R[l]-l+1,r-l+1);
            ans=max(ans,min(r-L[r]+1,r-l+1));
            ans=max(ans,RMQ(R[l]+1,L[r]-1));
            printf("%d\n",ans);
        }
    }
}