Files
2025-02-Numerical/lib/nr/ansi/recipes/select.c
2025-09-12 18:55:25 +09:00

46 lines
781 B
C

#define SWAP(a,b) temp=(a);(a)=(b);(b)=temp;
float select(unsigned long k, unsigned long n, float arr[])
{
unsigned long i,ir,j,l,mid;
float a,temp;
l=1;
ir=n;
for (;;) {
if (ir <= l+1) {
if (ir == l+1 && arr[ir] < arr[l]) {
SWAP(arr[l],arr[ir])
}
return arr[k];
} else {
mid=(l+ir) >> 1;
SWAP(arr[mid],arr[l+1])
if (arr[l] > arr[ir]) {
SWAP(arr[l],arr[ir])
}
if (arr[l+1] > arr[ir]) {
SWAP(arr[l+1],arr[ir])
}
if (arr[l] > arr[l+1]) {
SWAP(arr[l],arr[l+1])
}
i=l+1;
j=ir;
a=arr[l+1];
for (;;) {
do i++; while (arr[i] < a);
do j--; while (arr[j] > a);
if (j < i) break;
SWAP(arr[i],arr[j])
}
arr[l+1]=arr[j];
arr[j]=a;
if (j >= k) ir=j-1;
if (j <= k) l=i;
}
}
}
#undef SWAP