HihoCoder - 1710 等差子数列 【RMQ】

给定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
依次对于每个询问输出一个整数，代表答案。

Sample Input
6 2  
1 2 3 5 7 9  
2 6  
1 4
Sample Output
4  
3

题解：定义一个b[ i ] 数组，记录从 i 开始往前数最长的等差序列，dp[ i ][ j ] 表示从第 i 个数开始数 2^j 个数中的最大值，w[ i ][ j ] 表示从第 i 数开始数2^j 个数中最大值的下标；

所以可以知道 dp[ i ][ 0 ]=b[ i ], w[ i ][ 0 ]=i;

dp[ i ][ j ]=max(dp[ i ][ j-1 ]，dp[ i+(1<<(j-1)) ][ j-1 ])，如果两区间的最大值相同，w 记录后面区间的下标；

 例如：

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

 

dp[i][0]	1	2	3	2	3
dp[i][1]	2	3	3	3
dp[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<cstdio>
#include<cstdlib>
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<math.h>
#include<map>
#include<set>
#include<queue>
#include<vector>
#include<stack>
#define inf 0x3f3f3f3f
using namespace std;
typedef long long ll;
const double pi=acos(-1.0);
const ll mod=1e9+7;
const int N=1e5+10;

int dp[N][20];
int a[N],b[N],w[N][20];

void advan(int n)
{
    for(int i=1; i<=n; i++)
    {
        dp[i][0]=b[i];
        w[i][0]=i;
    }
    for(int j=1; (1<<j)<=n; j++)
    {
        for(int i=1; i+(1<<j)-1<=n; i++)
        {
            if(dp[i][j-1]>dp[i+(1<<(j-1))][j-1])
            {
                dp[i][j]=dp[i][j-1];
                w[i][j]=w[i][j-1];
            }
            else
            {
                dp[i][j]=dp[i+(1<<(j-1))][j-1];
                w[i][j]=w[i+(1<<(j-1))][j-1];
            }
        }
    }
}

int RMQ(int l,int r,int flag)
{
    int k=0;
    while((1<<(k+1))<=r-l+1) k++;
    int u,ans;
    if(dp[l][k]>dp[r-(1<<k)+1][k])
    {
        ans=dp[l][k];
        u=w[l][k];
    }
    else
    {
        ans=dp[r-(1<<k)+1][k];
        u=w[r-(1<<k)+1][k];
    }
    if(flag) return ans;//如果第二次进行RMQ，直接反正ans
    if(u-l+1>=ans) return ans;
    if(u==r) return r-l+1;
    ans=u-l+1;
    int anss=RMQ(u+1,r,1);//进行第二次查找
    return max(ans,anss);
}

int main()
{
    int l,r,n,m;
    while(~scanf("%d %d",&n,&m))
    {
        b[1]=1;
        int k=inf;//k表示前一个数和后一个数的差
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            if(i>=2)
            {
                if(a[i]-a[i-1]==k)
                    b[i]=b[i-1]+1;
                else
                {
                    k=a[i]-a[i-1];
                    b[i]=2;
                }
            }
        }
        advan(n);
        for(int i=0; i<m; i++)
        {
            scanf("%d %d",&l,&r);
            printf("%d\n",RMQ(l,r,0));
        }
    }
    return 0;
}