bzoj2124 等差子序列（树状数组+hash）

题意

给你一个1~n排列，问有没有一个等差数列（长度至少为3）

题解

我居然自己想到了正解。

但我最后写挂了，所以我又看了题解。

我们维护了一个以权值为下标的01序列。

我们扫描整个序列。对于每一个正在扫描的数，我们判断以这个数的权值作为对称点，01序列是否对称。

这个序列用权值树状数组维护就行。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#define LL long long 
const LL mod=1e9+7;
int n,t;
const int N=10010;
LL pw[N],c1[N],c2[N];
int a[N];
int min(int a,int b){
    if(a<b)return a;
    else return b;
}
int lowbit(int x){
    return x&(-x);
}
LL check1(int x){
    LL ans=0;
    for(int i=x;i>=1;i-=lowbit(i)){
        ans=(ans+(c1[i]*pw[x-i])%mod)%mod;
    }
    return ans;
}
LL check2(int x){
    LL ans=0;
    for(int i=x;i>=1;i-=lowbit(i)){
        ans=(ans+(c2[i]*pw[x-i])%mod)%mod;
    }
    return ans;
}
LL add1(int x){
       for(int i=x;i<=n;i+=lowbit(i)){
           c1[i]=(c1[i]+pw[i-x])%mod;
    }
}
LL add2(int x){
    for(int i=x;i<=n;i+=lowbit(i)){
        c2[i]=(c2[i]+pw[i-x])%mod;
    }
    return 0;
}
LL query1(int l,int r){
    LL p=check1(l-1),q=check1(r);
    return ((q-p*pw[r-l+1])%mod+mod)%mod;
}
LL query2(int l,int r){
    LL p=check2(l-1),q=check2(r);
    return ((q-p*pw[r-l+1])%mod+mod)%mod;
}
int main(){
    scanf("%d",&t);
    pw[0]=1;
    for(int i=1;i<=10001;i++)pw[i]=(pw[i-1]*(LL)2)%mod;
       for(int z=1;z<=t;z++){
        scanf("%d",&n);
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        memset(c1,0,sizeof(c1));memset(c2,0,sizeof(c2));
        for(int i=1;i<=n;i++){
            int x=a[i];
            int len=min(n-x,x-1);
            if(len&&query1(x-len,x-1)!=query2(n-x-len+1,n-x)){printf("Y
");break;}
            add1(x);add2(n-x+1);
            if(i==n)printf("N
");
        }
    }
    return 0;
}