codevs1283等差子序列

链接:http://codevs.cn/problem/1283/

题意:中文题。

分析:题目要求我们找是否存在等差数列,其实我们可以将条件弱化为是否存在长度为3的等差数列,直接暴力找会超时(但是这个OJ上的数据好弱,暴力可过),我们可以将这n个数慢慢加入到一个计数数组中,比如说加入x我们就令f[x]=1,这样的话我们在加入x的时候只要判断1~x-1是否和x+1~n是否对称(以x为中心对称)。这样就变成了两段字符串的判重了,hash就好了,更新的话我们将这个f数组建线段树,每次logn更新和合并即可。判重的时候也是logn。O(nlogn)

代码:

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=10010;
const int MAX=151;
const int mod=100000000;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=998244353;
const ll INF=10000000010;
typedef double db;
typedef unsigned long long ull;
int a[N];
ll ha[N],l[4*N],r[4*N];
void updata(int w,int le,int ri,int x) {
    if (le+1==ri) {
        l[w]=r[w]=1ull;return ;
    }
    int mid=(le+ri)/2;
    if (x<mid) updata(w<<1,le,mid,x);
    else updata((w<<1)+1,mid,ri,x);
    l[w]=(l[w<<1]+l[(w<<1)+1]*ha[mid-le]%MOD)%MOD;
    r[w]=(r[w<<1]*ha[ri-mid]%MOD+r[(w<<1)+1])%MOD;
}
ll getsum(int w,int L,int R,int le,int ri,int rev,int LL,int RR) {
    if (le>=ri) return 0;
    if (L==le&&R==ri) {
        if (rev>0) return l[w]*ha[le-LL]%MOD;
        else return r[w]*ha[RR-ri]%MOD;
    }
    int mid=(L+R)/2;
    if (ri<=mid) return getsum(w<<1,L,mid,le,ri,rev,LL,RR);
    else if (le>=mid) return getsum((w<<1)+1,mid,R,le,ri,rev,LL,RR);
    else return (getsum(w<<1,L,mid,le,mid,rev,LL,RR)+getsum((w<<1)+1,mid,R,mid,ri,rev,LL,RR))%MOD;
}
int main()
{
    int i,n,t,bo;
    ha[0]=1;for (i=1;i<N;i++) ha[i]=ha[i-1]*3%MOD;
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);
        for (i=1;i<=n;i++) scanf("%d", &a[i]);
        memset(l,0,sizeof(l));
        memset(r,0,sizeof(r));
        bo=0;
        for (i=1;i<=n;i++) {
            if (a[i]<=(n+1)/2&&getsum(1,1,n+1,1,a[i],1,1,a[i])!=getsum(1,1,n+1,a[i]+1,2*a[i],-1,a[i]+1,2*a[i])) bo=1;
            if (2*a[i]>n&&getsum(1,1,n+1,2*a[i]-n,a[i],1,2*a[i]-n,a[i])!=getsum(1,1,n+1,a[i]+1,n+1,-1,a[i]+1,n+1)) bo=1;
            if (bo) break ;
            updata(1,1,n+1,a[i]);
        }
        if (bo) printf("Y\n");
        else printf("N\n");
    }
    return 0;
}


你可能感兴趣的:(codevs1283等差子序列)