HihoCoder - 1710 等差子数列( rmq )

给定N个整数A₁, A₂, ... A_N，小Hi会询问你M个问题。

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

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

Input

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

第二行包含N个整数A₁, A₂, ... A_N。  

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

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

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

Output

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

Sample Input

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

Sample Output

4  
3

解：我们有一个d[i] 数组 ，存从i点出发，往前推的等差连续子序列为多长 ，然后我们的ma[i][j]表示d数组中从第i个数起连续2^j个数中的最大值，w[i][j]表示相应最大值的位置 （DP的状态） 

显然  ma[i][0]=d[i]   w[i][0]=i (初始化)

状态转移方程ma[i][ j]=max（ma[i][j-1], ma[i + 2^(j-1)][j-1]）     ,如果两者相同w[i][j]=w[i+2^(j-1)][j-1] 

 

a[i]	1	2	3	6	9
d[i]	1	2	3	2	3

ma[i][0]	1	2	3	2	3
ma[i][1]	2	3	3	3
ma[i][2]	3	3

w[i][0]	1	2	3	4	5
w[i][1]	2	3	3	5
w[i][2]	3	5

然后，我们开始查询每一个  l 和 r  ，首先我们找出 最大值ans 和最大值的地址 u  ，如果 u-l+1>=ans ,说明一整个等差数列都在范围内，输出ans

如果 u=r ，那么直接输出 u-l+1 ,   否则的话  ans=max（u-l+1 ，anss）；anss代表 （u+1，r）范围内的最大值  显然anss代表的数列肯定在

（l，r）范围内

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<iostream>
#include<math.h>
#include<queue>
#include<map>
#include<vector>
#include<set>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
int a[100010];
int d[100010];
int ma[100010][30];
int w[100010][30];
int n,m;
void aa()//预处理
{
    for(int i=1;i<=n;i++)
    {
        ma[i][0]=d[i];
        w[i][0]=i;
    }
    for(int j=1;(1<<j)<=n;j++)
        for(int i=1;i+(1<<j)-1<=n;i++)
        {
          if(ma[i][j-1]>ma[i+(1<<(j-1))][j-1])
          {
              ma[i][j]=ma[i][j-1];
              w[i][j]=w[i][j-1];
          }
          else//如果两者相同 取后面的，因为后面的肯定在范围内
          {
              ma[i][j]=ma[i+(1<<(j-1))][j-1];
              w[i][j]=w[i+(1<<(j-1))][j-1];
          }
        }
}
int rmq(int l,int r,int flag)//flag代表第一次还是第二次，如果是第二次那么直接输出区间内最大值即可
{
    int k=0;
    while((1<<(k+1))<=r-l+1) k++; 
    int u,ans;
    if(ma[l][k]>ma[r - (1 << k) + 1][k])
    {
        ans=ma[l][k];
        u=w[l][k];
    }
    else
    {
        ans=ma[r - (1 << k) + 1][k];
        u=w[r - (1 << k) + 1][k];
    }
    if(flag) return ans;//如果是第二次,输出最大值,这个值代表anss
    if(u-l+1>=ans)
        return ans;
    if(u==r)
        return u-l+1;
    ans=u-l+1;
    int anss=rmq(u+1,r,1); //进行第二次查找
    return max(ans,anss);
}
int main()
{
    scanf("%d%d",&n,&m);
    int k=inf,l,r;//k代表后一个数和前一个数的差值
    d[1]=1;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        if(i==1) continue;
        if(a[i]-a[i-1]==k)
           d[i]=d[i-1]+1;
        else
        {
            k=a[i]-a[i-1];
            d[i]=2;//最小是2
        }
    }
    aa();
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&l,&r);
        int ans=rmq(l,r,0);
        printf("%d\n",ans);
    }

}
3 1
1 2 3
2 3

2

4 1
1 2 3 1
3 4

2

10 1
1 2 3 4 5 6 8 10 12 14
5 10

5