bzoj2124 等差子序列

Description

给一个1到N的排列{Ai}，询问是否存在1<=p1<p2<p3<p4<p5<…<pLen<=N (Len>=3)，

使得Ap1,Ap2,Ap3,…ApLen是一个等差序列。

Input

输入的第一行包含一个整数T，表示组数。

下接T组数据，每组第一行一个整数N，每组第二行为一个1到N的排列，数字两两之间用空格隔开。

N<=10000，T<=7

Output

对于每组数据，如果存在一个等差子序列，则输出一行“Y”，否则输出一行“N”。

Sample Input

2
3
1 3 2
3
3 2 1

Sample Output

N
Y

只要找到一个长度为3的就可以了

也就是j<i<k,a[i]-a[j]=a[k]-a[i]

也就是a[i]-x和a[i]+x在i不同侧

b[a[i]]=1表示在前面出现过，0表示没有

因为是排列，所以等价于b[a[i]-x]!=b[a[i]+x]

把它看作一个以a[i]为中心的最长串，如果不是回文串，那么显然存在答案

用树状数组维护字符串hash，正向反向维护2个

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef unsigned long long ull; 
const int N=10007;
const int p=10007;
ull bin[N],c1[N],c2[N];
int n,a[N];
void add1(int x)
{int i;
    for (i=x;i<=n;i+=(i&(-i)))
    c1[i]+=bin[i-x];
}
void add2(int x)
{int i;
    for (i=x;i<=n;i+=(i&(-i)))
    c2[i]+=bin[i-x];
}
ull query1(int x)
{ull s=0;
int i;
    for (i=x;i;i-=(i&(-i)))
    s+=c1[i]*bin[x-i];
    return s;
}
ull query2(int x)
{ull s=0;
int i;
    for (i=x;i;i-=(i&(-i)))
    s+=c2[i]*bin[x-i];
    return s;
}
ull cal1(int l,int r)
{
    ull h1=query1(l-1),h2=query1(r);
    return h2-bin[r-l+1]*h1;
}
ull cal2(int l,int r)
{
    ull h1=query2(l-1),h2=query2(r);
    return h2-bin[r-l+1]*h1;
}
int main()
{int T,i,flag;
    cin>>T;
    bin[0]=1;
    for (i=1;i<N;i++)
    bin[i]=bin[i-1]*p;
    while (T--)
    {
        cin>>n;
        memset(c1,0,sizeof(c1));
        memset(c2,0,sizeof(c2));
        for (i=1;i<=n;i++)
        {
            scanf("%d",&a[i]); 
        }
        flag=0;
        for (i=1;i<=n;i++)
        {
            int L=min(a[i]-1,n-a[i]);
            if (L&&cal1(a[i]-L,a[i]-1)!=cal2(n-a[i]-L+1,n-a[i]))
            {flag=1;break;}
            add1(a[i]);
            add2(n-a[i]+1);
        }
        if (flag) cout<<"Y"<<endl;
        else cout<<"N"<<endl;
    }
}