Cod sursa(job #379949)

Utilizator mlazariLazari Mihai mlazari Data 4 ianuarie 2010 14:16:38
Problema Statistici de ordine Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 0.71 kb
#include<stdio.h>
#include<time.h>
#include<stdlib.h>


#define NMAX 3000003

int n,k,i,j,l,r,d,sol,m[NMAX];

void swap(int &a,int &b) {
  int c=a;
  a=b;
  b=c;
}

int randPiv(int l,int r) {
  return l+rand()%(r-l+1);
}

int main() {
  freopen("sdo.in","r",stdin);
  freopen("sdo.out","w",stdout);
  scanf("%d%d",&n,&k);
  for(i=1;i<=n;i++) scanf("%d",m+i);
  l=1;
  r=n;
  do {
    i=l;
    j=r;
    swap(m[i],m[randPiv(l,r)]);
    d=0;
    while(i<j) {
      if(m[i]>m[j]) {
        swap(m[i],m[j]);
        d=1-d;
      }
      i+=d;
      j-=1-d;
    }
    if(i==k) sol=m[k];
    else
     if(k<i) r=i-1;
     else l=i+1;
  } while(!sol);
  printf("%d\n",sol);
  return 0;
}