Choose m elements randomly from a vector containing n elements

Question

I have a vector containing n elements. I need to choose a subset of m elements randomly from the vector without repetition. What is the most efficient way of doing this? I need to do this several thousands of times in my code.

The solution on top of my mind is to use rand() to generate a random number k between 0 and n. Then pick the kth element in the vector and insert it into a std::set. Keep doing this till the set's size becomes equal to m. I am now assured that the set contains m unique elements randomly chosen from the set of n elements.

What are the other possible solutions?

Thanks.

Answer 1

You want a Fisher-Yates shuffle (stop after M iterations):

template<class BidiIter > BidiIter random_unique(BidiIter begin, BidiIter end, size_t num_random) {     size_t left = std::distance(begin, end);     while (num_random--) {         BidiIter r = begin;         std::advance(r, rand()%left);         std::swap(*begin, *r);         ++begin;         --left;     }     return begin; } 

Demo at http://ideone.com/3A3cv. This is significantly faster than std::random_shuffle when you only need a few random numbers out of the set, and should be just about the same speed even if N==M.