Description
给出一个长度为n的排列
a
a
,求是否存在一个三元组(i,j,k)满足:且
ai<aj<ak
a
i
<
a
j
<
a
k
输出YES或NO
Solution
注意:题目只要求判断是否存在方案,
显然,如果答案为NO,那么对于每一个位置i,所以满足
aj<ai<ak
a
j
<
a
i
<
a
k
的j,k必须同时小于或大于i,也就是在i的同侧,
暴力可以遍历左侧的每一个位置x,当:
1.
ax<ai
a
x
<
a
i
时,如果
ai−ax)≤min(ai,n−ai)
a
i
−
a
x
)
≤
min
(
a
i
,
n
−
a
i
)
且
ai+(ai−ax)
a
i
+
(
a
i
−
a
x
)
这个数在i右侧,那么答案为YES;
2.
ax>ai
a
x
>
a
i
时,如果
ax−ai≤min(ai,n−ai)
a
x
−
a
i
≤
min
(
a
i
,
n
−
a
i
)
且
ai−(ax−ai)
a
i
−
(
a
x
−
a
i
)
这个数在i右侧,那么答案为YES,
那么,我们把i左侧的数排序后,取
ai−min(ai,n−ai)
a
i
−
min
(
a
i
,
n
−
a
i
)
到
ai+min(ai,n−ai)
a
i
+
min
(
a
i
,
n
−
a
i
)
之间的数,判断是否沿
ai
a
i
对称即可,
如: 3,5,2,4,6,1 i=4,
取左侧2~6的数,3,5是对称的的,但2不是,所以为YES
这个可以用线段树+hash实现,
复杂度: O(nlog(n)) O ( n log ( n ) )
Code
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cmath>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
#define efo(i,q) for(int i=A[q];i;i=B[i][0])
using namespace std;
typedef long long LL;
const int N=300500,mo=998244353;
int read(int &n)
{
char ch=' ';int q=0,w=1;
for(;(ch!='-')&&((ch<'0')||(ch>'9'));ch=getchar());
if(ch=='-')w=-1,ch=getchar();
for(;ch>='0' && ch<='9';ch=getchar())q=q*10+ch-48;n=q*w;return n;
}
int m,n,ans;
LL a[N];
bool OK;
struct qqww
{
LL co,sum,v;
}b[N*4],bZ;
qqww merge(qqww q,qqww w)
{
qqww ans;
ans.co=q.co+w.co;
ans.sum=q.sum+w.sum;
ans.v=q.v+w.v;
return ans;
}
void change(int l,int r,int e,int l1)
{
if(l==r)
{
b[e].co=1;
b[e].sum=l;
b[e].v=(LL)l*l;
return;
}
int t=(l+r)>>1;
if(l1<=t)change(l,t,e*2,l1);
else change(t+1,r,e*2+1,l1);
b[e]=merge(b[e*2],b[e*2+1]);
}
qqww find(int l,int r,int e,int l1,int r1)
{
if(r1<l1)return bZ;
if(l1<=l&&r<=r1)return b[e];
int t=(l+r)>>1;
qqww ans=bZ;
if(l1<=t)ans=find(l,t,e*2,l1,r1);
if(t<r1)ans=merge(ans,find(t+1,r,e*2+1,l1,r1));
return ans;
}
int main()
{
freopen("poem.in","r",stdin);
freopen("poem.out","w",stdout);
int q;
read(n);
fo(i,1,n)a[i]=read(q);
OK=0;
fo(i,1,n)
{
change(1,n,1,a[i]);
q=min(a[i]-1,n-a[i]);
qqww t=find(1,n,1,a[i]-q,a[i]-1);
qqww t1=find(1,n,1,a[i]+1,a[i]+q);
if(t.co!=t1.co||a[i]*t.co-t.sum!=t1.sum-t1.co*(LL)a[i]||t.v-2LL*a[i]*t.sum!=t1.v-2LL*a[i]*t1.sum)
{OK=1;break;}
}
if(OK)printf("YES\n");
else printf("NO\n");
return 0;
}