Mercurial > repos > alg_tools / file revision

///! This module implements some arithmetic functions that are valid in a ‘const context”, i.e.,
///! can be computed at compile-time.

/// Calculate $n!$.
/// (The version in the crate `num_integer` deesn't currently work in a `const`  context.
pub const fn factorial(n : usize) -> usize {
    const fn f(i : usize, a : usize) -> usize {
        if i==0 { a } else { f(i-1, i*a) }
    }
    f(n, 1)
}

/// Calculate $\nchoosek{n}{k}$.
/// (The version in the crate `num_integer` deesn't currently work in a `const` context.
pub const fn binomial(n : usize, k : usize) -> usize {
    factorial(n)/(factorial(n-k)*factorial(k))
}

/// Calculate $n^k$.
/// (The version in the crate `num_integer` deesn't currently work in a `const` context.
pub const fn pow(n : usize, k : usize) -> usize {
    const fn pow_accum(n : usize, k : usize, accum : usize) -> usize {
        if k==0 { accum } else { pow_accum(n, k-1, accum*n) }
    }
    pow_accum(n, k, 1)
}