我们可以使用math库中的perm函数来计算n个项目的k排列数量。perm函数的语法如下:
math.perm(n, k)
以下是代码示例:
import math
def count_permutations(n, k):
"""
计算n个项目的k排列数量
"""
return math.perm(n, k)
我们可以使用以下代码测试函数的输出:
print(count_permutations(5, 2)) # 输出:20
print(count_permutations(6, 3)) # 输出:120
print(count_permutations(4, 4)) # 输出:24
以上代码将计算5个项目的2排列数量(应为20)、6个项目的3排列数量(应为120)和4个项目的4排列数量(应为24)。