BZOJ 2124 等差子序列（线段树）

Description
给一个$1$到$N$的排列${A_i}$,询问是否存在$1\leq p_1<p_2<p_3<p_4<p_5<\cdots<p_{Len}\leq N$$(3\leq len)$,使得$A_{p_1},A_{p_2},A_{p_3},\cdots,A_{p_{Len}}$是一个等差序列

Input
输入的第一行包含一个整数$T$,表示组数.
下接$T$组数据,每组第一行一个整数$N$,每组第二行为一个$1$到$N$的排列,数字两两之间用空格隔开

Output
对于每组数据,如果存在一个等差子序列,则输出一行”Y”，否则输出一行”N”.

Sample Input
2
3
1 3 2
3
3 2 1

Sample Output
N
Y

Data Size & Hint
$N\leq 10000,T\leq 7$

Solution

要求一个$len\geq 3$的等差子序列,那还不如直接求$len=3$的等差子序列

考虑等差子序列的性质$a_1+a_3=2a_2$
设中项为$x$,易得左侧有一点$x+d$,右侧有一点$x-d$,$1 < abs(d) < x$

如果$x+d,x-d$没有出现在$x$的同一侧,那么显然的,必然会出现$len=3$的等差子序列
考虑维护一侧的数(没有出现的就在另一侧)

当前询问到第$i$个位置,为$p_i$
在$1-n$数轴上标记已经出现过的数
每次从数轴上点$p_i$开始向两端扩展$(l=r=i,l--,r++)$

若$p_i$为中心两端对称分布$(0-0,1-1)$(忽略多余部分),则说明$p_i+d,p_i-d$都出现在$p_i$的同一侧(全0即右侧,全1即左侧),不存在以$i$位置为中心的等差子序列
否则存在以$i$位置为中心的等差子序列

考虑如何维护左右部分是否对称

要求:单点修改,区间求和
树状数组或线段树即可

开两个同时维护正向与反向,每次提取一段对比即可
$lenth=min(n-a[i],a[i]-1);$
正向区间$a[i],a[i]+lenth$
反向区间$n-a[i]+1,n-a[i]+1+lenth$
若相等,则不存;若不等,则存在

单点修改线段树直接维护就行
记得update比较特殊,需要乘上一个相对位置

具体看代码.

代码如下:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define L (x<<1)
#define R (x<<1)+1
#define mid ((l+r)>>1)
const int N = 100010;
const ll mod = 19260817;
int n,a[N];
ll p[N];
struct tree {
    ll node[N<<4],siz[N];
    void clear() {
        memset(node,0,sizeof(node));
        memset(siz,0,sizeof(siz));
    }
    void update(int x) {node[x]=(node[L]*p[siz[R]]%mod+node[R])%mod;}
    void build(int x,int l,int r) {
        if(l!=r) {
            build(L,l,mid);
            build(R,mid+1,r);
            siz[x]=siz[L]+siz[R];
        }
        else siz[x]=1;
    }
    void change(int x,int p,int l,int r) {
        if(l==r) {node[x]=1;return;}
        if(p<=mid) change(L,p,l,mid);
        else change(R,p,mid+1,r);
        update(x);
    }
    ll query(int x,int l,int r,int _l,int _r) {
        if(l>=_l && r<=_r) {return node[x]*p[_r-r]%mod;}
        ll ans=0;
        if(mid>=_l) ans+=query(L,l,mid,_l,_r);
        if(mid<_r) ans+=query(R,mid+1,r,_l,_r);
        ans%=mod;
        return ans;
    }
}t1,t2;
int read() {
    int ans=0,flag=1;
    char ch=getchar();
    while( (ch>'9' || ch<'0') && ch!='-' ) ch=getchar();
    if(ch=='-') {flag=-1;ch=getchar();}
    while(ch>='0' && ch<='9') {ans=ans*10+ch-'0';ch=getchar();}
    return ans*flag;
}
void init() {
    n=read();p[0]=1;
    t1.clear();t2.clear();
    t1.build(1,1,n);t2.build(1,1,n);
    for(int i=1;i<=n;++i) {
        a[i]=read();
        p[i]=p[i-1]*2%mod;
    }
}
void work() {
    bool flag=0;
    for(int i=1;i<=n;++i) {
        t1.change(1,a[i],1,n);
        t2.change(1,n-a[i]+1,1,n);
        int lenth=min(n-a[i],a[i]-1);
        if(lenth && t1.query(1,1,n,a[i],a[i]+lenth) != t2.query(1,1,n,n-a[i]+1,n-a[i]+1+lenth)) {
            puts("Y");
            flag=1;
            break;
        }
    }
    if(!flag) puts("N");
}
int main() {
    int t=read();
    while(t--) {
        init();
        work();
    }
    return 0;
}