codevs1283等差子序列

来源：互联网发布：ipadmini无法安装软件编辑：程序博客网时间：2024/05/16 18:46

链接：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;}

1 0