大致题意: 给定一个\(n\)的排列,问是否存在\(i<j<k\)满足\(p_j=\frac{p_i+p_k}2\)。
转化
注意到这是一个排列。
不妨设\(q_x\)表示数\(x\)所在的位置,问题就变成是否存在\(x,t\)满足\(q_{x-t}<q_x<q_{x+t}\)或\(q_{x-t}>q_x>q_{x+t}\)。
如果我们按照\(q_x\)从小到大枚举\(x\)(即按\(i\)从小到大枚举\(x=p_i\)),把其余的\(q\)根据与\(q_x\)的大小关系修改为\(0/1\),则问题就变成对于每个\(x\)是否存在\(t\)满足\(q_{x-t}\not=q_{x+t}\)。
那么对于\(x\)若不存在一个合法的\(t\),相当于从\(x\)出发两侧的字符完全相等。
只要把\(q\)中的信息哈希起来用线段树维护,就可以快速比较了。
代码
#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define Reg register
#define RI Reg int
#define Con const
#define CI Con int&
#define I inline
#define W while
#define N 300000
using namespace std;
int n,p[N+5];
struct Hash//哈希
{
#define ull unsigned long long
#define CU Con ull&
ull x,y;I Hash() {x=y=0;}I Hash(CU a) {x=y=a;}I Hash(CU a,CU b):x(a),y(b){}
I Hash operator + (Con Hash& o) Con {return Hash(x+o.x,y+o.y);}
I Hash operator - (Con Hash& o) Con {return Hash(x-o.x,y-o.y);}
I Hash operator * (Con Hash& o) Con {return Hash(x*o.x,y*o.y);}
I bool operator != (Con Hash& o) Con {return x^o.x||y^o.y;}
}seed(19,302627441),P[N+5];
class SegmentTree//线段树维护哈希值
{
private:
#define PT CI l=1,CI r=n,CI rt=1
#define LT l,mid,rt<<1
#define RT mid+1,r,rt<<1|1
#define PU(x) (A[x]=A[x<<1]*P[r-mid]+A[x<<1|1],B[x]=B[x<<1]+B[x<<1|1]*P[mid-l+1])
Hash A[N<<2],B[N<<2];
public:
I void U(CI x,PT)//单点修改
{
if(l==r) return (void)(A[rt]=B[rt]=1);RI mid=l+r>>1;
x<=mid?U(x,LT):U(x,RT),PU(rt);
}
I Hash QA(CI L,CI R,PT)//正着
{
if(L==l&&r==R) return A[rt];RI mid=l+r>>1;if(R<=mid) return QA(L,R,LT);
if(L>mid) return QA(L,R,RT);return QA(L,mid,LT)*P[R-mid]+QA(mid+1,R,RT);
}
I Hash QB(CI L,CI R,PT)//倒着
{
if(L==l&&r==R) return B[rt];RI mid=l+r>>1;if(R<=mid) return QB(L,R,LT);
if(L>mid) return QB(L,R,RT);return QB(L,mid,LT)+QB(mid+1,R,RT)*P[mid-L+1];
}
}S;
int main()
{
RI i;for(scanf("%d",&n),P[0]=i=1;i<=n;++i) scanf("%d",p+i),P[i]=P[i-1]*seed;//读入同时,预处理哈希种子的幂
RI t;for(i=1;i<=n;S.U(p[i]),++i) if(t=min(p[i]-1,n-p[i]))//枚举数
if(S.QA(p[i]-t,p[i]-1)!=S.QB(p[i]+1,p[i]+t)) return puts("YES");//判断两侧01是否相同
return puts("NO"),0;
}