Question

以下是spoj问题http://www.spoj.com/problems/GOODB/的链接。你被赋予N, WA, TLE, RE

N = WA+TLE+RE

有多少不同的N不正确结果的有序组合（每个都是WA，TLE或RE），它们满足了它们的预测？

这是我在python中的工作解决方案

import math
mod=10**9+7
def nck(n,k):
    return math.factorial(n)/(math.factorial(n-k) * math.factorial(k))

n,w,t,r = map(int, raw_input().split())
print (nck(n,w) * nck(n-w, t)) % mod

这是针对相同问题的第二种方法，假设排列方式的数量 n1 a1，n2 a2，...，nk ak连续是

n!/(n1! n2! .... nk !)

，其中

n = n1 + n2 +...+ nk

这是我的第二种方法的代码

def modexp(x, y, mod):
    res = 1
    if x==0 or x==1:
        return 1
    while y != 0:
        if y & 1 == 1:
            'if y is odd'
            res = (res * x) % mod
        x = (x * x) % mod
        y >>= 1
   return res

def modfact(n, mod):
    res = 1
    while n >= 1:
        res = (res * n) % mod
        n -= 1
    return res

mod = 10 ** 9 + 7
n, w, t, r = map(int, input().split())
resn = modfact(n, mod)
resw = modexp(modfact(w, mod), mod - 2, mod)
rest = modexp(modfact(t, mod), mod - 2, mod)
resr = modexp(modfact(r, mod), mod - 2, mod)
res = (resn*resw*rest*resr)%mod
print(res)

我只是无法弄清楚为什么我的第二种方法是错误的。如果有人提供任何有关我出错的见解。

Answer 1

这一行很可疑：

res = resn//(resw*rest*resr)

resw和朋友是阶乘的模块化逆，所以它们应该相乘而不是分割。最终结果应该由10 ** 9 + 7修改。

N的有序组合有多少种

1 个答案: