【JZOJ 5557】写诗

本文介绍了一种高效算法,用于判断给定排列中是否存在满足特定条件的三元组(i,j,k)。通过使用线段树和哈希技术,该算法能够在O(nlog(n))的时间复杂度内完成任务。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Description

给出一个长度为n的排列aa,求是否存在一个三元组(i,j,k)满足:i<j<kai<aj<akai<aj<ak
输出YES或NO

Solution

注意:题目只要求判断是否存在方案,

显然,如果答案为NO,那么对于每一个位置i,所以满足aj<ai<akaj<ai<ak的j,k必须同时小于或大于i,也就是在i的同侧,
暴力可以遍历左侧的每一个位置x,当:
1. ax<aiax<ai时,如果aiax)min(ai,nai)ai−ax)≤min(ai,n−ai)ai+(aiax)ai+(ai−ax)这个数在i右侧,那么答案为YES;
2. ax>aiax>ai时,如果axaimin(ai,nai)ax−ai≤min(ai,n−ai)ai(axai)ai−(ax−ai)这个数在i右侧,那么答案为YES,
那么,我们把i左侧的数排序后,取aimin(ai,nai)ai−min(ai,n−ai)ai+min(ai,nai)ai+min(ai,n−ai)之间的数,判断是否沿aiai对称即可,
如: 3,5,2,4,6,1 i=4,
取左侧2~6的数,3,5是对称的的,但2不是,所以为YES

这个可以用线段树+hash实现,

复杂度:O(nlog(n))O(nlog⁡(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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值