Description
给一个1到N的排列{Ai},询问是否存在1<=p1<p2<p3<p4<p5<…<pLen<=N (Len>=3),
使得Ap1,Ap2,Ap3,…ApLen是一个等差序列。
输入的第一行包含一个整数T,表示组数。
下接T组数据,每组第一行一个整数N,每组第二行为一个1到N的排列,数字两两之间用空格隔开。
N<=10000,T<=7
Solution
考虑暴力怎么做,我们枚举中项
a
j
a_j
aj,枚举首项
a
i
(
i
≤
j
)
a_i (i\leq j )
ai(i≤j),用桶判断是否存在
a
k
=
2
a
j
−
a
i
(
j
≤
k
)
a_k=2a_j-a_i (j\leq k)
ak=2aj−ai(j≤k),这样是Tn2的
可以发现我们枚举j的时候和做卷积是一样的,考虑用bitset优化
记前缀桶为A,后缀桶为B。对于a[i]我们在A中a[i]的位置插入1,在B中20000-a[i]的位置插入1
若B>>(20000-a[i])&A不全为0说明存在这样三个等差数列,于是这样做是
T
n
2
32
T\frac{n^2}{32}
T32n2的
题解做法:我们开一个桶b[],依次插入a。若a[i]是中项则桶中一定存在j使得b[j]!=b[2*i-j]。
我们把b看成字符串,a[i]为中项即b不存在以i为中心的回文子串,这个东西可以哈希判断。我们要资瓷插入、求区间哈希,于是上线段树就可以了
Code
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <bitset>
#define rep(i,st,ed) for (register int i=st;i<=ed;++i)
#define fill(x,t) memset(x,t,sizeof(x))
typedef long long LL;
const int MOD=2147483647;
const int N=10005;
LL sum[N<<2][2],wjp[N];
int a[N];
inline int read() {
int x=0,v=1; char ch=getchar();
for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
return x*v;
}
void modify(int now,int tl,int tr,int x) {
if (tl==tr) return (void) (sum[now][0]=sum[now][1]=1);
int mid=(tl+tr)>>1;
if (x<=mid) modify(now<<1,tl,mid,x);
else modify(now<<1|1,mid+1,tr,x);
sum[now][0]=(1LL*sum[now<<1][0]*wjp[tr-mid]%MOD+sum[now<<1|1][0])%MOD;
sum[now][1]=(1LL*sum[now<<1|1][1]*wjp[mid-tl+1]%MOD+sum[now<<1][1])%MOD;
}
LL query(int now,int tl,int tr,int l,int r,int opt) {
if (r<l) return 0;
if (tl==l&&tr==r) return sum[now][opt];
int mid=(tl+tr)>>1;
if (r<=mid) return query(now<<1,tl,mid,l,r,opt);
if (l>mid) return query(now<<1|1,mid+1,tr,l,r,opt);
int qx=query(now<<1,tl,mid,l,mid,opt);
int qy=query(now<<1|1,mid+1,tr,mid+1,r,opt);
if (!opt) return (1LL*qx*wjp[3,r-mid]%MOD+qy)%MOD;
return (1LL*qy*wjp[mid-l+1]%MOD+qx)%MOD;
}
void build(int now,int tl,int tr) {
sum[now][0]=sum[now][1]=0;
if (tl==tr) return ;
int mid=(tl+tr)>>1;
build(now<<1,tl,mid); build(now<<1|1,mid+1,tr);
}
int main(void) {
wjp[0]=1; rep(i,1,N-1) wjp[i]=wjp[i-1]*3LL%MOD;
for (int T=read();T--;) {
int n=read(),flag=false;
build(1,1,n);
rep(i,1,n) a[i]=read();
rep(i,1,n) {
int len=std:: min(a[i],n-a[i]+1);
LL qx=query(1,1,n,a[i]-len+1,a[i]-1,0);
LL qy=query(1,1,n,a[i]+1,a[i]+len-1,1);
if (qx!=qy) {
// printf("%d\n", i);
flag=true;
break;
}
modify(1,1,n,a[i]);
}
(!flag)?puts("N"):puts("Y");
}
return 0;
}