算法训练 多阶乘计算

资源限制

时间限制：1.0s   内存限制：256.0MB

问题描述

　　我们知道，阶乘n!表示n*(n-1)*(n-2)*......*2*1, 类似的，可以定义多阶乘计算，例如：5！！=5*3*1,依次可以有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

 

理解了题意以后直接写

m=2时计算并输出n(1)!+n(2)!+......+n(k)!的各个位上的数字之和。：

这句话的意思是对很多个n的阶乘加起来的结果 的各个位 求和得到的和

例：5 1 2

5的一阶乘=5*4*3*2*1=120，1+2=3，输出3

碰到阶乘一定要考虑数据范围用不用longlong

#include<iostream>
#include<cstring>
using namespace std;

long long multi(int n,int k){
	long long res=1;
	while(n>0){
		res*=n;
		n-=k;
	}
	return res;
}
int main(){
	int n,k,m;
	long long sum=0;
	cin>>n>>k>>m;
	for(int i=1;i<=k;i++){
		sum+=multi(n,i);

	}
	if(m==1){
		cout<<sum<<endl;
	}
	else{
		int res=0;
		while(sum){
			res+=sum%10;
			sum/=10;
		}
		cout<<res<<endl;
	}
	return 0;
}