试题 算法训练 多阶乘计算

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　我们知道，阶乘n!表示n*(n-1)(n-2)…21, 类似的，可以定义多阶乘计算，例如：5！！=531,依次可以有n!..!(k个‘！’，可以简单表示为n(k)!)=n*(n-k)(n-2k)…（直到最后一个数<=0）。
　　现给定一组数据n、k、m,当m=1时，计算并输出n(1)!+n(2)!+…+n(k)!的值，m=2时计算并输出n(1)!+n(2)!+…+n(k)!的各个位上的数字之和。
输入格式
　　两行，第一行为n和k，第二行为m。
输出格式
　　一行,为n(1)!+n(2)!+…+n(k)!的值或n(1)!+n(2)!+…+n(k)!的各个位上的数字之和。
样例输入
5 1
2
样例输出
3
数据规模和约定
　　0 < k < n <= 20
这题说的模模糊糊 醉了 md

"""
@Author:Lixiang

@Blog(个人博客地址): https://lixiang007.top/

@WeChat：18845312866

"""
import math
import string
import sys
import cmath
from itertools import permutations
n,k=list(map(int,input().strip().split()))
m=int(input())
sum=1
temp=0
for i in range(1,k+1):
    sum = 1
    j=n
    while j>0:
        sum*=j
        j -= i
    temp+=sum
if m==2:
    str1=str(temp)
    res=0
    for i in str1:
        res+=int(i)
    print(res)
else:
    print(temp)