寻找与bigz兼容的组合函数

时间:2013-10-04 13:23:43

标签: r combinatorics biginteger

我正在捣乱一些数论&想获得bigz兼容版本的combn(计算comb(n,j)的函数)。 bigzgmp包生成和处理的扩展整数的类名 我知道我可以将bigz转换为double并使用“整数作为浮点数”,至少在我得到一些非常大的整数之前,但是如果有人写了一些与{{{{1}兼容的组合函数1}},我很欣赏指针。 (否则我会离开并尝试自己编写:-()

1 个答案:

答案 0 :(得分:0)

好的,所以这是与combn一起运行的修改后的bigz代码。请注意,输入验证不完整,特别是尚未实现FUN函数具有bigz方法的验证。我的最终版本,我将转发给gmp的维护者,(我希望!!)将实现它作为“combin.bigz”方法,而不是一个独立的函数。

combnz <- function (x, m, FUN = NULL, simplify = TRUE, ...) 
{
    stopifnot(length(m) == 1L)
    if (m < 0) 
        stop("m < 0", domain = NA)
    # Really not much point in creating a sequence 1:gigundo_bigz
    #if (is.numeric(x) && length(x) == 1L && x > 0 && trunc(x) ==  x) 
        #x <- seq_len(x)
    n <- length(x)
    if (n < m) 
        stop("n < m", domain = NA)
    m <- as.integer(m)
    e <- 0
    h <- m
    a <- seq_len(m)
    nofun <- is.null(FUN)
    if (!nofun && !is.function(FUN)) 
        stop("'FUN' must be a function or NULL")
# TODO: add a " grep('bigz',methods(FUN) )" to verify there's a method.
    len.r <- length(r <- if (nofun) x[a] else FUN(x[a], ...))
    count <- as.integer(round(choose(n, m)))
    if (simplify) {
        dim.use <- if (nofun) 
            c(m, count)
        else {
            d <- dim(r)
            if (length(d) > 1L) 
                c(d, count)
            else if (len.r > 1L) 
                c(len.r, count)
            else c(d, count)
        }
    }
    if (simplify) {
        out <- matrix(r, nrow = len.r, ncol = count)
    }
    else {
        out <- vector("list", count)
        out[[1L]] <- r
    }
    if (m > 0) {
        i <- 2L
        nmmp1 <- n - m + 1L
        while (a[1L] != nmmp1) {
            if (e < n - h) {
                h <- 1L
                e <- a[m]
                j <- 1L
            }
            else {
                e <- a[m - h]
                h <- h + 1L
                j <- 1L:h
            }
            a[m - h + j] <- e + j
            r <- if (nofun) 
                x[a]
            else FUN(x[a], ...)
            if (simplify)
# bigz doesn't  handle replacement the way regular matrices do.  So...
                out[1:len.r,i] <- r
                # out[, i] <- r
# I suspect a list element ain't gonna care
            else out[[i]] <- r
            i <- i + 1L
        }
    }
    if (simplify) 
        array(out, dim.use)
    else out
}