我正在捣乱一些数论&想获得bigz
兼容版本的combn
(计算comb(n,j)的函数)。 bigz
是gmp
包生成和处理的扩展整数的类名
我知道我可以将bigz
转换为double
并使用“整数作为浮点数”,至少在我得到一些非常大的整数之前,但是如果有人写了一些与{{{{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
}