Problem
Write a method to randomly generate a set of m integers from an array of size n Each element must have equal probability of being chosen.
Solution
This question depends on 2 things:Now that we have fisher yates algorithm in our quiver, we can solve this problem :
let m be the number of elements to select for i = 1; i <= m; i++ pick a random number from 1 to n, call it j swap array[j] and array [n] (assuming 1 indexed arrays) n--
public static int[] chooseNFromM(int[] array, int m) {
if (m > array.length)
return null;
int[] results = new int[m];
for (int i = 0; i < m; ++i) {
int index, temp;
do {
index = new Random().nextInt(array.length);
} while (index < i);
temp = array[index];
array[index] = array[i];
array[i] = temp;
}
for (int i = 0; i < m; ++i) {
results[i] = array[i];
}
return results;
}
We don't need to keep generating random numbers. We just need to label which ones are 'dead' (in this case, 0 to i) and generate random index in the valid range (i+1 to n).
We should first make an copy of the original array as we don't want to modify it.
References







0 comments:
Post a Comment