Randomized Algorithms | Set 3 (1/2 Approximate Median)
We strongly recommend to refer below articles as a prerequisite of this.
Randomized Algorithms | Set 1 (Introduction and Analysis)
Randomized Algorithms | Set 2 (Classification and Applications)
In this post, a Monte Carlo algorithm is discussed.
Problem Statement : Given an unsorted array A[] of n numbers and ε > 0, compute an element whose rank (position in sorted A[]) is in the range [(1 – ε)n/2, (1 + ε)n/2].
For ½ Approximate Median Algorithm &epsilom; is 1/2 => rank should be in the range [n/4, 3n/4]
We can find k’th smallest element in O(n) expected time and O(n) worst case time.
What if we want in less than O(n) time with low probable error allowed?
Following steps represent an algorithm that is O((Log n) x (Log Log n)) time and produces incorrect result with probability less than or equal to 2/n2.
- Randomly choose k elements from the array where k=c log n (c is some constant)
- Insert then into a set.
- Sort elements of the set.
- Return median of the set i.e. (k/2)th element from the set
C
#include<bits/stdc++.h>
using namespace std;
int randApproxMedian( int arr[], int n)
{
random_device rand_dev;
mt19937 generator(rand_dev());
uniform_int_distribution< int > distribution(0, n-1);
if (n==0)
return 0;
int k = 10*log2(n);
set< int > s;
for ( int i=0; i<k; i++)
{
int index = distribution(generator);
s.insert(arr[index]);
}
set< int > ::iterator itr = s.begin();
advance(itr, (s.size()/2) - 1);
return *itr;
}
int main()
{
int arr[] = {1, 3, 2, 4, 5, 6, 8, 7};
int n = sizeof (arr)/ sizeof ( int );
printf ( "Approximate Median is %d\n" ,randApproxMedian(arr,n));
return 0
}
|
Java
import java.util.Iterator;
import java.util.Random;
import java.util.TreeSet;
class Test
{
static int arr[] = new int []{ 1 , 3 , 2 , 4 , 5 , 6 , 8 , 7 } ;
static int randApproxMedian( int n)
{
Random r = new Random();
if (n== 0 )
return 0 ;
double k1 = 10 *Math.log(n);
int k = ( int )k1;
TreeSet s = new TreeSet<Integer>();
for ( int i= 0 ; i<k; i++)
{
int index = r.nextInt(n);
s.add(arr[index]);
}
Iterator<Integer> itr = s.iterator();
int temp = s.size()/ 2 - 1 ;
for ( int i = 0 ; i < temp; i++) {
itr.next();
}
return itr.next();
}
public static void main(String[] args) {
System.out.println( "Approximate Median is " + randApproxMedian(arr.length));
}
}
|
Output:
Approximate Median is 4
Time Complexity:
We use a set provided by the STL in C++. In STL Set, insertion for each element takes O(log k). So for k insertions, time taken is O (k log k).
Now replacing k with c log n
=>O(c log n (log (clog n))) =>O (log n (log log n))
How is probability of error less than 2/n2?
Algorithm makes an error if the set S has at least k/2 elements are from the Left Quarter or Right Quarter.
It is quite easy to visualize this statement since the median which we report will be (k/2)th element and if we take k/2 elements from the left quarter(or right quarter) the median will be from the left quarter (or the right quarter).
An array can be divided into 4 quarters each of size n/4. So P(selecting left quarter) is 1/4. So what is the probability that at least k/2 elements are from the Left Quarter or Right Quarter? This probability problem is same as below :
Given a coin which gives HEADS with probability 1/4 and TAILS with 3/4. The coin is tossed k times. What is the probability that we get at least k/2 HEADS is less than or equal to?
Explanation:
If we put k = c log n for c = 10, we get
P <= (1/2)2log n
P <= (1/2)log n2
P <= n-2
Probability of selecting at least k/2 elements from the left quarter) <= 1/n2
Probability of selecting at least k/2 elements from the left or right quarter) <= 2/n2
Therefore algorithm produces incorrect result with probability less than or equal to 2/n2.
References:www.cse.iitk.ac.in/users/sbaswana/CS648/Lecture-2-CS648.pptx
Last Updated :
27 Mar, 2023
Like Article
Save Article
Share your thoughts in the comments
Please Login to comment...