我正在使用一些数论并想获得一个bigz
兼容版本的combn
(计算 comb(n,j) 的函数)。 bigz
是gmp
包生成和处理的扩展整数的类名。
我知道我可以摆脱转换bigz
并double
使用“整数作为浮点数”,至少在我得到一些非常大的整数之前,但如果有人编写了一些与 兼容的组合函数gmp
,我会很感激一个指针。(否则我会去尝试自己写它们:-()
问问题
67 次
1 回答
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
}
于 2013-10-04T15:34:43.767 回答