From 5dd7f82a34501dc48b985c9fe4822cfa2ef125c2 Mon Sep 17 00:00:00 2001
From: realstealthninja <68815218+realstealthninja@users.noreply.github.com>
Date: Fri, 13 Sep 2024 15:39:18 +0530
Subject: [PATCH] chore: use iwyu on probablity/**.cpp

---
 probability/geometric_dist.cpp  | 17 +++++++++--------
 probability/windowed_median.cpp | 13 +++++++------
 2 files changed, 16 insertions(+), 14 deletions(-)

diff --git a/probability/geometric_dist.cpp b/probability/geometric_dist.cpp
index a9bc7aeb6..16a8edfb2 100644
--- a/probability/geometric_dist.cpp
+++ b/probability/geometric_dist.cpp
@@ -19,14 +19,15 @@
  * @author [Domenic Zingsheim](https://github.com/DerAndereDomenic)
  */
 
-#include <cassert>   /// for assert
-#include <cmath>     /// for math functions
-#include <cstdint>   /// for fixed size data types
-#include <ctime>     /// for time to initialize rng
-#include <iostream>  /// for std::cout
-#include <limits>    /// for std::numeric_limits
-#include <random>    /// for random numbers
-#include <vector>    /// for std::vector
+#include <bits/std_abs.h>  // for abs
+#include <stdlib.h>        // for rand, srand, RAND_MAX
+#include <cassert>         // for assert
+#include <cmath>           // for log, pow, sqrt
+#include <cstdint>         // for uint32_t
+#include <ctime>           // for time
+#include <iostream>        // for basic_ostream, char_traits, operator<<, cout
+#include <limits>          // for numeric_limits
+#include <vector>          // for vector
 
 /**
  * @namespace probability
diff --git a/probability/windowed_median.cpp b/probability/windowed_median.cpp
index 52c70ae24..5998d90f4 100644
--- a/probability/windowed_median.cpp
+++ b/probability/windowed_median.cpp
@@ -28,12 +28,13 @@
  * Time complexity: O(logN). Space complexity: O(N). N - size of window
  * @author [Yaniv Hollander](https://github.com/YanivHollander)
  */
-#include <cassert>  /// for assert
-#include <cstdlib>  /// for std::rand - needed in testing
-#include <ctime>    /// for std::time - needed in testing
-#include <list>     /// for std::list - used to manage sliding window
-#include <set>  /// for std::multiset - used to manage multi-value sorted sliding window values
-#include <vector>  /// for std::vector - needed in testing
+#include <cassert>   // for assert
+#include <cstdlib>   // for rand, RAND_MAX, srand
+#include <ctime>     // for time
+#include <list>      // for _List_iterator, list
+#include <set>       // for _Rb_tree_const_iterator, multiset
+#include <vector>    // for vector
+#include <iterator>  // for next
 
 /**
  * @namespace probability