rustgym 0.2.0

rustgym solutions
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use rand::distributions::WeightedIndex;
use rand::prelude::*;

struct Solution {
    dist: WeightedIndex<i32>,
    rng: ThreadRng,
}

impl Solution {
    fn new(w: Vec<i32>) -> Self {
        let rng = thread_rng();
        let dist = WeightedIndex::new(w).unwrap();
        Solution { dist, rng }
    }

    fn pick_index(&mut self) -> i32 {
        self.rng.sample(&self.dist) as i32
    }
}