2124: 等差子序列 - BZOJ

  1 const
  2     maxn=10010;
  3     h=1000000007;
  4 type
  5     node=record
  6       l,r,lc,rc:longint;
  7       h1,h2:int64;
  8     end;
  9 var
 10     f:array[0..maxn*2]of node;
 11     a:array[0..maxn]of longint;
 12     p:array[0..maxn]of int64;
 13     t,n,tot:longint;
 14 
 15 function min(x,y:longint):longint;
 16 begin
 17     if x<y then exit(x);
 18     exit(y);
 19 end;
 20 
 21 function max(x,y:longint):longint;
 22 begin
 23     if x>y then exit(x);
 24     exit(y);
 25 end;
 26 
 27 procedure add(x,now:longint);
 28 var
 29     mid:longint;
 30 begin
 31     if f[now].l=f[now].r then
 32     begin
 33       f[now].h1:=1;
 34       f[now].h2:=1;
 35       exit;
 36     end;
 37     mid:=(f[now].l+f[now].r)>>1;
 38     if x>mid then add(x,f[now].rc)
 39     else add(x,f[now].lc);
 40     f[now].h1:=(f[f[now].lc].h1*p[f[now].r-mid]+f[f[now].rc].h1)mod h;
 41     f[now].h2:=(f[f[now].rc].h2*p[mid-f[now].l+1]+f[f[now].lc].h2)mod h;
 42 end;
 43 
 44 procedure build(l,r:longint);
 45 var
 46     now,mid:longint;
 47 begin
 48     inc(tot);
 49     now:=tot;
 50     f[now].l:=l;
 51     f[now].r:=r;
 52     f[now].h1:=0;
 53     f[now].h2:=0;
 54     f[now].lc:=0;
 55     f[now].rc:=0;
 56     if l=r then exit;
 57     mid:=(l+r)>>1;
 58     f[now].lc:=tot+1;
 59     build(l,mid);
 60     f[now].rc:=tot+1;
 61     build(mid+1,r);
 62 end;
 63 
 64 function hash1(l,r,now:longint):int64;
 65 var
 66     mid:longint;
 67 begin
 68     if (l<=f[now].l) and (r>=f[now].r) then exit(f[now].h1);
 69     mid:=(f[now].l+f[now].r)>>1;
 70     if l>mid then exit(hash1(l,r,f[now].rc));
 71     if r<=mid then exit(hash1(l,r,f[now].lc));
 72     exit((hash1(l,r,f[now].lc)*p[min(r,f[now].r)-mid]+hash1(l,r,f[now].rc))mod h);
 73 end;
 74 
 75 function hash2(l,r,now:longint):int64;
 76 var
 77     mid:longint;
 78 begin
 79     if (l<=f[now].l) and (r>=f[now].r) then exit(f[now].h2);
 80     mid:=(f[now].l+f[now].r)>>1;
 81     if l>mid then exit(hash2(l,r,f[now].rc));
 82     if r<=mid then exit(hash2(l,r,f[now].lc));
 83     exit((hash2(l,r,f[now].rc)*p[mid-max(l,f[now].l)+1]+hash2(l,r,f[now].lc))mod h);
 84 end;
 85 
 86 procedure main;
 87 var
 88     i:longint;
 89 begin
 90     read(n);
 91     for i:=1 to n do
 92       read(a[i]);
 93     tot:=0;
 94     build(1,n);
 95     for i:=2 to n-1 do
 96       begin
 97         add(a[i-1],1);
 98         if hash1(a[i]-min(a[i]-1,n-a[i]),a[i],1)=hash2(a[i],a[i]+min(a[i]-1,n-a[i]),1) then continue;
 99         writeln(‘Y‘);
100         exit;
101       end;
102     writeln(‘N‘);
103 end;
104 
105 begin
106     p[0]:=1;
107     for t:=1 to maxn do
108       p[t]:=p[t-1]*2 mod h;
109     read(t);
110     while t>0 do
111       begin
112         dec(t);
113         main;
114       end;
115 end.