Bernoulli Sample — Interview Query algorithms problem