洛谷P2757 [国家集训队]等差子序列

题目描述

给一个1到N的排列{Ai}，询问是否存在

1 \le p_1<p_2<p_3<p_4<p_5<…<p_{Len} \le N (Len \ge 3)1≤p1​<p2​<p3​<p4​<p5​<…<pLen​≤N(Len≥3)

使得 Ap_1,Ap_2,Ap_3,\cdots,Ap_{Len}Ap1​,Ap2​,Ap3​,⋯,ApLen​ 是一个等差序列。

输入输出格式

输入格式：

输入的第一行包含一个整数T，表示组数。

下接T组数据，每组第一行一个整数N，每组第二行为一个1到N的排列，数字两两之间用空格隔开。

输出格式：

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

输入输出样例

输入样例#1：  复制

2
3
1 3 2
3
3 2 1

输出样例#1：  复制

N
Y

说明

对于5%的数据，N<=100

对于30%的数据，N<=1000

对于100%的数据，N<=10000，T<=7

题解：

显然只要判断有没有长度为3的就好了。

从左到右，设当前数x为中间那一个，显然已出现过的1..x-1和n..x+1如果有一位不相等（也就是它们的值不相等）那么就成立，因为一个在前，还有一个没出现过只能在后了。

线段树维护正反哈希就行。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
#define li unsigned long long
li d[5][440000],up[110000]; 
inline int read(){
      int x=0,f=1;char s=getchar();
      while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
      while(s>='0'&&s<='9'){x=(x<<3)+(x<<1)+(s-'0');s=getchar();}
      return x*f;
}
inline li q(int le,int ri,int x,int y,int k,int wh){
      if(le>=x&&ri<=y){
        return d[k][wh];
      }
      int mid=(le+ri)>>1;
      if(mid>=y)return q(le,mid,x,y,k,wh<<1);
        else if(mid<x)return q(mid+1,ri,x,y,k,wh<<1|1);
        else{
            if(!k){
              return q(le,mid,x,y,k,wh<<1)*up[min(ri,y)-mid]+q(mid+1,ri,x,y,k,wh<<1|1);
            }else {
              return q(mid+1,ri,x,y,k,wh<<1|1)*up[mid-max(le,x)+1]+q(le,mid,x,y,k,wh<<1);
          }
        }
}
inline void update(int le,int ri,int pl,int wh){
      if(le==ri){
        d[0][wh]=d[1][wh]=233;
        return;
      }
      int mid=(le+ri)>>1;
      if(mid>=pl)update(le,mid,pl,wh<<1);
        else update(mid+1,ri,pl,wh<<1|1);
      d[0][wh]=d[0][wh<<1]*up[ri-mid]+d[0][wh<<1|1];
      d[1][wh]=d[1][wh<<1|1]*up[mid-le+1]+d[1][wh<<1];
}
int main()
{     int n,m,x,i,j,k,t;
      up[0]=1;
      for(i=1;i<=10010;i++)up[i]=up[i-1]*233;
      t=read();
      while(t--){
          memset(d,0,sizeof(d));
          n=read();
          int ok=0;
          for(i=1;i<=n;i++){
              x=read();
              if(!ok){
                  update(1,n,x,1);
                  if(x!=1&&x!=n){
                    int len=min(x-1,n-x);
                    li f1=q(1,n,x-len,x-1,0,1);
                    li f2=q(1,n,x+1,x+len,1,1);
                    if(f1!=f2)ok=1;
                  }
              }
          }
          if(ok)puts("Y");
            else puts("N");
      }
      return 0;
}