Add WeightedIndexTree to dist_randr (#1372)

Add WeightedIndexTree to rand_distr Co-authored-by: xmakro <makro@>
rust-random · Feb 8, 2024 · f827c00 · f827c00
1 parent 1f4507a
commit f827c00
Show file tree

Hide file tree

Showing 3 changed files with 433 additions and 23 deletions.
diff --git a/rand_distr/src/lib.rs b/rand_distr/src/lib.rs
@@ -76,8 +76,9 @@
 //!   - [`UnitBall`] distribution
 //!   - [`UnitCircle`] distribution
 //!   - [`UnitDisc`] distribution
-//! - Alternative implementation for weighted index sampling
+//! - Alternative implementations for weighted index sampling
 //!   - [`WeightedAliasIndex`] distribution
+//!   - [`WeightedTreeIndex`] distribution
 //! - Misc. distributions
 //!   - [`InverseGaussian`] distribution
 //!   - [`NormalInverseGaussian`] distribution
@@ -133,6 +134,9 @@ pub use rand::distributions::{WeightedError, WeightedIndex};
 #[cfg(feature = "alloc")]
 #[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
 pub use weighted_alias::WeightedAliasIndex;
+#[cfg(feature = "alloc")]
+#[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
+pub use weighted_tree::WeightedTreeIndex;
 
 pub use num_traits;
 
@@ -186,6 +190,9 @@ mod test {
 #[cfg(feature = "alloc")]
 #[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
 pub mod weighted_alias;
+#[cfg(feature = "alloc")]
+#[cfg_attr(doc_cfg, doc(cfg(feature = "alloc")))]
+pub mod weighted_tree;
 
 mod binomial;
 mod cauchy;