Package stdlib
Class StdRandom
java.lang.Object
stdlib.StdRandom
This class provides static methods for generating random number from various discrete and continuous distributions,
including uniform, Binomial, Bernoulli, geometric, Gaussian, exponential, Pareto, Poisson, and Cauchy. It also
provides method for shuffling an array or subarray and generating random permutations.
By convention, all intervals are half open. For example,
uniform(-1.0, 1.0)
returns a random number between
-1.0
(inclusive) and 1.0
(exclusive).
Similarly, shuffle(a, lo, hi)
shuffles the hi - lo
elements in the array a[]
, starting at index lo
(inclusive) and ending at index hi
(exclusive).
For additional documentation, see Section 2.2 of Computer Science: An Interdisciplinary Approach by Robert Sedgewick and Kevin Wayne.
- Author:
- Robert Sedgewick, Kevin Wayne
-
Method Summary
Modifier and TypeMethodDescriptionstatic boolean
Returns a random boolean from a Bernoulli distribution with success probability 1/2.static boolean
bernoulli
(double p) Returns a random boolean from a Bernoulli distribution with success probability p.static int
binomial
(int n, double p) Flips a coin with biasp
a total ofn
times and returns the number of heads obtained.static double
cauchy()
Returns a random real number from the Cauchy distribution.static int
discrete
(double[] probabilities) Returns a random integer from the specified discrete distribution.static int
discrete
(int[] frequencies) Returns a random integer from the specified discrete distribution.static double
exp
(double lambda) Returns a random real number from an exponential distribution with rate λ.static double
gaussian()
Returns a random real number from a standard Gaussian distribution.static double
gaussian
(double mu, double sigma) Returns a random real number from a Gaussian distribution with mean μ and standard deviation σ.static int
geometric
(double p) Returns a random integer from a geometric distribution with success probability p.static long
getSeed()
Returns the seed of the pseudo-random number generator.static void
Unit tests the methods in this class.static double
pareto()
Returns a random real number from the standard Pareto distribution.static double
pareto
(double alpha) Returns a random real number from a Pareto distribution with shape parameter α.static int[]
permutation
(int n) Returns a uniformly random permutation of n elements.static int[]
permutation
(int n, int k) Returns a uniformly random permutation of k of n elements.static int
poisson
(double lambda) Returns a random integer from a Poisson distribution with mean λ.static void
setSeed
(long s) Sets the seed of the pseudo-random number generator.static void
shuffle
(double[] a) Rearranges the elements of the specified array in uniformly random order.static void
shuffle
(double[] a, int lo, int hi) Rearranges the elements of the specified subarray in uniformly random order.static void
shuffle
(int[] a) Rearranges the elements of the specified array in uniformly random order.static void
shuffle
(int[] a, int lo, int hi) Rearranges the elements of the specified subarray in uniformly random order.static void
Rearranges the elements of the specified array in uniformly random order.static void
Rearranges the elements of the specified subarray in uniformly random order.static double
uniform()
Returns a random real number uniformly in [0, 1).static double
uniform
(double a, double b) Returns a random real number uniformly in [a, b).static int
uniform
(int n) Returns a random integer uniformly in [0, n).static int
uniform
(int a, int b) Returns a random integer uniformly in [a, b).static long
uniform
(long n) Returns a random long integer uniformly in [0, n).
-
Method Details
-
setSeed
public static void setSeed(long s) Sets the seed of the pseudo-random number generator. This method enables you to produce the same sequence of "random" number for each execution of the program. Ordinarily, you should call this method at most once per program.- Parameters:
s
- the seed
-
getSeed
public static long getSeed()Returns the seed of the pseudo-random number generator.- Returns:
- the seed
-
uniform
public static double uniform()Returns a random real number uniformly in [0, 1).- Returns:
- a random real number uniformly in [0, 1)
-
uniform
public static int uniform(int n) Returns a random integer uniformly in [0, n).- Parameters:
n
- number of possible integers- Returns:
- a random integer uniformly between 0 (inclusive) and
n
(exclusive) - Throws:
IllegalArgumentException
- ifn <= 0
-
uniform
public static long uniform(long n) Returns a random long integer uniformly in [0, n).- Parameters:
n
- number of possiblelong
integers- Returns:
- a random long integer uniformly between 0 (inclusive) and
n
(exclusive) - Throws:
IllegalArgumentException
- ifn <= 0
-
uniform
public static int uniform(int a, int b) Returns a random integer uniformly in [a, b).- Parameters:
a
- the left endpointb
- the right endpoint- Returns:
- a random integer uniformly in [a, b)
- Throws:
IllegalArgumentException
- ifb <= a
IllegalArgumentException
- ifb - a >= Integer.MAX_VALUE
-
uniform
public static double uniform(double a, double b) Returns a random real number uniformly in [a, b).- Parameters:
a
- the left endpointb
- the right endpoint- Returns:
- a random real number uniformly in [a, b)
- Throws:
IllegalArgumentException
- unlessa < b
-
binomial
public static int binomial(int n, double p) Flips a coin with biasp
a total ofn
times and returns the number of heads obtained.- Parameters:
n
- number of coin flipsp
- bias- Returns:
- number of heads obtained
-
bernoulli
public static boolean bernoulli(double p) Returns a random boolean from a Bernoulli distribution with success probability p.- Parameters:
p
- the probability of returningtrue
- Returns:
true
with probabilityp
andfalse
with probability1 - p
- Throws:
IllegalArgumentException
- unless0
≤p
≤1.0
-
bernoulli
public static boolean bernoulli()Returns a random boolean from a Bernoulli distribution with success probability 1/2.- Returns:
true
with probability 1/2 andfalse
with probability 1/2
-
gaussian
public static double gaussian()Returns a random real number from a standard Gaussian distribution.- Returns:
- a random real number from a standard Gaussian distribution (mean 0 and standard deviation 1).
-
gaussian
public static double gaussian(double mu, double sigma) Returns a random real number from a Gaussian distribution with mean μ and standard deviation σ.- Parameters:
mu
- the meansigma
- the standard deviation- Returns:
- a real number distributed according to the Gaussian distribution
with mean
mu
and standard deviationsigma
-
geometric
public static int geometric(double p) Returns a random integer from a geometric distribution with success probability p. The integer represents the number of independent trials before the first success.- Parameters:
p
- the parameter of the geometric distribution- Returns:
- a random integer from a geometric distribution with success
probability
p
; orInteger.MAX_VALUE
ifp
is (nearly) equal to1.0
. - Throws:
IllegalArgumentException
- unlessp >= 0.0
andp <= 1.0
-
poisson
public static int poisson(double lambda) Returns a random integer from a Poisson distribution with mean λ.- Parameters:
lambda
- the mean of the Poisson distribution- Returns:
- a random integer from a Poisson distribution with mean
lambda
- Throws:
IllegalArgumentException
- unlesslambda > 0.0
and not infinite
-
pareto
public static double pareto()Returns a random real number from the standard Pareto distribution.- Returns:
- a random real number from the standard Pareto distribution
-
pareto
public static double pareto(double alpha) Returns a random real number from a Pareto distribution with shape parameter α.- Parameters:
alpha
- shape parameter- Returns:
- a random real number from a Pareto distribution with shape
parameter
alpha
- Throws:
IllegalArgumentException
- unlessalpha > 0.0
-
cauchy
public static double cauchy()Returns a random real number from the Cauchy distribution.- Returns:
- a random real number from the Cauchy distribution.
-
discrete
public static int discrete(double[] probabilities) Returns a random integer from the specified discrete distribution.- Parameters:
probabilities
- the probability of occurrence of each integer- Returns:
- a random integer from a discrete distribution:
i
with probabilityprobabilities[i]
- Throws:
IllegalArgumentException
- ifprobabilities
isnull
IllegalArgumentException
- if sum of array entries is not (very nearly) equal to1.0
IllegalArgumentException
- unlessprobabilities[i] >= 0.0
for each indexi
-
discrete
public static int discrete(int[] frequencies) Returns a random integer from the specified discrete distribution.- Parameters:
frequencies
- the frequency of occurrence of each integer- Returns:
- a random integer from a discrete distribution:
i
with probability proportional tofrequencies[i]
- Throws:
IllegalArgumentException
- iffrequencies
isnull
IllegalArgumentException
- if all array entries are0
IllegalArgumentException
- iffrequencies[i]
is negative for any indexi
IllegalArgumentException
- if sum of frequencies exceedsInteger.MAX_VALUE
(231 - 1)
-
exp
public static double exp(double lambda) Returns a random real number from an exponential distribution with rate λ.- Parameters:
lambda
- the rate of the exponential distribution- Returns:
- a random real number from an exponential distribution with
rate
lambda
- Throws:
IllegalArgumentException
- unlesslambda > 0.0
-
shuffle
Rearranges the elements of the specified array in uniformly random order.- Parameters:
a
- the array to shuffle- Throws:
IllegalArgumentException
- ifa
isnull
-
shuffle
public static void shuffle(double[] a) Rearranges the elements of the specified array in uniformly random order.- Parameters:
a
- the array to shuffle- Throws:
IllegalArgumentException
- ifa
isnull
-
shuffle
public static void shuffle(int[] a) Rearranges the elements of the specified array in uniformly random order.- Parameters:
a
- the array to shuffle- Throws:
IllegalArgumentException
- ifa
isnull
-
shuffle
Rearranges the elements of the specified subarray in uniformly random order.- Parameters:
a
- the array to shufflelo
- the left endpoint (inclusive)hi
- the right endpoint (exclusive)- Throws:
IllegalArgumentException
- ifa
isnull
IllegalArgumentException
- unless(0 <= lo) && (lo < hi) && (hi <= a.length)
-
shuffle
public static void shuffle(double[] a, int lo, int hi) Rearranges the elements of the specified subarray in uniformly random order.- Parameters:
a
- the array to shufflelo
- the left endpoint (inclusive)hi
- the right endpoint (exclusive)- Throws:
IllegalArgumentException
- ifa
isnull
IllegalArgumentException
- unless(0 <= lo) && (lo < hi) && (hi <= a.length)
-
shuffle
public static void shuffle(int[] a, int lo, int hi) Rearranges the elements of the specified subarray in uniformly random order.- Parameters:
a
- the array to shufflelo
- the left endpoint (inclusive)hi
- the right endpoint (exclusive)- Throws:
IllegalArgumentException
- ifa
isnull
IllegalArgumentException
- unless(0 <= lo) && (lo < hi) && (hi <= a.length)
-
permutation
public static int[] permutation(int n) Returns a uniformly random permutation of n elements.- Parameters:
n
- number of elements- Returns:
- an array of length
n
that is a uniformly random permutation of0
,1
, ...,n-1
- Throws:
IllegalArgumentException
- ifn
is negative
-
permutation
public static int[] permutation(int n, int k) Returns a uniformly random permutation of k of n elements.- Parameters:
n
- number of elementsk
- number of elements to select- Returns:
- an array of length
k
that is a uniformly random permutation ofk
of the elements from0
,1
, ...,n-1
- Throws:
IllegalArgumentException
- ifn
is negativeIllegalArgumentException
- unless0 <= k <= n
-
main
Unit tests the methods in this class.- Parameters:
args
- the command-line arguments
-