POJ2104 区间第k小

题意就是区间第k大……

题解：

前段时间用主席树搞掉了……

如今看到划分树，是在想来写一遍，结果18号对着学长的代码调了一上午连样例都没过，好桑心……

今天在做NOI2010超级钢琴，忽然发现用划分树很直观，果断决定再战划分树

对着网上的c++代码抄了一遍，A了，可是这编程复杂度有点高，忽然又看见盾哥的代码

很简短，和我原先的代码差不多，他怎么能A了呢……（后来发现区间第k小，我却写的第k大……sb啊）

考虑到盾哥的程序用到了离散化，因此没有考虑存在两个数相等的情况，这样就使代码减少很多

我报着试一试的心态把我认为肯定对的代码随机生成了几组大数据和盾哥的程序对拍，然后就对上了

好的，以后就用盾哥的程序

代码：

c++翻译来的：

 var  s,t:array[..,..] of longint;
      a,rk:array[..] of longint;
      i,n,m,x,y,k,j:longint;
  procedure sort(l,r:longint);
  var i,j,m,temp:longint;
  begin
  i:=l;j:=r;m:=a[(i+j)>>];
  repeat
   while a[i]<m do inc(i);
   while a[j]>m do dec(j);
   if i<=j then
    begin
    temp:=a[i];a[i]:=a[j];a[j]:=temp;
    inc(i);dec(j);
    end;
  until i>j;
  if i<r then sort(i,r);
  if j>l then sort(l,j);
  end;
   procedure init;
    begin
      readln(n,m);
      for i:= to n do read(a[i]);t[]:=a;
      sort(,n);
    end;
  procedure build(h,l,r:longint);
   var  mid,i,lp,rp,lm:longint;
   begin
   mid:=(l+r)>>;lm:=mid-l+;lp:=l;rp:=mid+;
   for i:=l to mid do
    if a[i]<a[mid] then dec(lm);
   for i:=l to r do
    begin
    if i=l then s[h,i]:= else s[h,i]:=s[h,i-];
    if t[h,i]=a[mid] then
     begin
     if lm<> then
      begin
      dec(lm);inc(s[h,i]);t[h+,lp]:=t[h,i];inc(lp);
      end
     else begin t[h+,rp]:=t[h,i];inc(rp);end;
     end
    else
     if t[h,i]<a[mid] then begin inc(s[h,i]);t[h+,lp]:=t[h,i];inc(lp);end
    else begin t[h+,rp]:=t[h,i];inc(rp);end;
    end;
   if l=r then exit;
   build(h+,l,mid);
   build(h+,mid+,r);
   end;
  function find(h,l,r,x,y,k:longint):longint;
   var mid,ll,rr:longint;
   begin
   if l=r then exit(t[h,l]);
   mid:=(l+r)>>;
   if l=x then ll:= else ll:=s[h,x-];rr:=s[h,y]-ll;
   if rr>=k then exit(find(h+,l,mid,l+ll,l+ll+rr-,k))
   else exit(find(h+,mid+,r,mid++x-l-ll,mid++y-l-ll-rr,k-rr));
   end;
  procedure main;
   begin
    build(,,n);
    for i:= to m do
     begin
      readln(x,y,k);
      writeln(find(,,n,x,y,k));
     end;
   end;
 begin
   init;
   main;
 end.       

盾哥的：

 program poj2104; const maxn=;
 var
   a,b:array[..maxn] of longint;
   s,d:array[..,..maxn] of longint;
   n,m,i,j,x,y,z,ls,rs,mid,p:longint;

 procedure sort(l,r:longint); var i,j,x,y:longint;
 begin
   i:=l;j:=r;x:=a[(i+j)>> ];
   repeat
     while b[a[i]]<b[x] do inc(i);
     while b[a[j]]>b[x] do dec(j);
     if i<=j then begin
       y:=a[i];a[i]:=a[j];a[j]:=y;
       inc(i);dec(j);
     end;
   until i>j;
   if l<j then sort(l,j);
   if i<r then sort(i,r);
 end;

 procedure buildtree(h,l,r:longint);var mid:longint;
 begin
   if l=r then exit;
   mid:=(l+r+)>> ;p:=;
   for i:=l to r do
     if d[h,i]<mid then begin
       d[h+,l+p]:=d[h,i];
       inc(p);s[h,i]:=p;
     end else begin
       d[h+,mid+i-l-p]:=d[h,i];
       s[h,i]:=p;
     end;
   buildtree(h+,l,mid-);
   buildtree(h+,mid,r);
 end;

 function find(h,ll,rr,l,r,k:longint):longint;
 begin
   if l=r then exit(d[h,r]);
   if l=ll then ls:= else ls:=s[h,l-];
   rs:=s[h,r];mid:=(ll+rr+)>> ;
   if rs-ls<k
     then find:=find(h+,mid,rr,l-ll-ls+mid,r-ll-rs+mid,k-rs+ls)
     else find:=find(h+,ll,mid-,ls+ll,rs+ll-,k);
 end;

 begin
   readln(n,m);
   for i:= to n do begin read(b[i]);a[i]:=i; end;
   sort(,n);
   for i:= to n do d[,a[i]]:=i;
   buildtree(,,n);
   for i:= to m do begin
     readln(x,y,z);
     writeln(b[a[find(,,n,x,y,z)]]);
   end;
 end.   

我最先抄的：

 const maxn=+;
 type arrtype=array[..maxn] of longint;
 var s,z:array[..,..maxn] of longint;
     a,b,rk:arrtype;
     i,j,n,m,x,y,k:longint;
 procedure sort(var a:arrtype;l,r:longint);
  var i,j,m,temp:longint;
  begin
  i:=l;j:=r;m:=a[(i+j)>>];
  repeat
   while a[i]<m do inc(i);
   while a[j]>m do dec(j);
   if i<=j then
    begin
    temp:=a[i];a[i]:=a[j];a[j]:=temp;
    temp:=rk[i];rk[i]:=rk[j];rk[j]:=temp;
    inc(i);dec(j);
    end;
  until i>j;
  if i<r then sort(a,i,r);
  if j>l then sort(a,l,j);
  end;
 procedure init;
  begin
  readln(n,m);
  for i:= to n do read(a[i]);readln;b:=a;
  for i:= to n do rk[i]:=i;
  sort(a,,n);
  end;
 procedure build(h,l,r:longint);
  var i,p,mid:longint;
  begin
  mid:=(l+r)>>;p:=;
  for i:=l to r do
   if z[h,i]<=mid then
    begin
     z[h+,l+p]:=z[h,i];
     inc(p);
     s[h,i]:=p;
    end
   else
    begin
     z[h+,mid++i-p-l]:=z[h,i];
     s[h,i]:=p;
    end;
  if l=r then exit;
  build(h+,l,mid);
  build(h+,mid+,r);
  end;
 function find(h,l,r,x,y,k:longint):longint;
  var ll,rr,mid:longint;
  begin
  if l=r then exit(z[h,l]);
  mid:=(l+r)>>;
  if l=x then ll:= else ll:=s[h,x-];rr:=s[h,y];
  if rr-ll>=k then exit(find(h+,l,mid,l+ll,l+rr-,k))
  else exit(find(h+,mid+,r,mid++x-l-ll,mid++y-l-rr,k-(rr-ll)));
  end;
 procedure main;
  begin
  for i:= to n do z[,rk[i]]:=i;
  build(,,n);
  for i:= to m do
   begin
   readln(x,y,k);
   writeln(b[rk[find(,,n,x,y,k)]]);
   end;
  end;
 begin
  init;
  main;
 end.        

盾哥的代码风格确实比我好很多，值得学习

UPD:

c++

 #include<cstdio>
 #include<cmath>
 #include<cstring>
 #include<algorithm>
 #include<iostream>
 #define inf 1000000000
 #define maxn 100000+100
 #define maxm 100000
 using namespace std;
 int n,m,a[maxn],sa[maxn],t[][maxn],s[][maxn];
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }
 void build(int h,int l,int r)
 {
     int mid=(l+r)>>,num=;
     for(int i=l;i<=r;i++)
     if (t[h][i]<=mid)
     {
         t[h+][l+num]=t[h][i];
         s[h][i]=++num;
     }
     else
     {
         t[h+][mid++i-l-num]=t[h][i];
         s[h][i]=num;
     }
     if(l==r)return;
     build(h+,l,mid);
     build(h+,mid+,r);
 }
 int find(int h,int l,int r,int x,int y,int k)
 {
     if(l==r)return t[h][l];
     int mid=(l+r)>>,ll,rr;
     ll=(l==x)?:s[h][x-];rr=s[h][y];
     if(rr-ll>=k) return find(h+,l,mid,l+ll,l+rr-,k);
     else return find(h+,mid+,r,mid++x-l-ll,mid++y-l-rr,k-(rr-ll));
 }
 bool cmp(int x,int y)
 {
     return a[x]<a[y];
 }
 int main()
 {
     freopen("input.txt","r",stdin);
     freopen("output.txt","w",stdout);
     n=read();m=read();
     for(int i=;i<=n;i++){a[i]=read();sa[i]=i;}
     sort(sa+,sa+n+,cmp);
     for(int i=;i<=n;i++)t[][sa[i]]=i;
     build(,,n);
     for(int i=;i<=m;i++)
     {
         int x,y,k;
         x=read();y=read();k=read();
         printf("%d\n",a[sa[find(,,n,x,y,k)]]);
     }
     return ;
 }