蓝桥杯 算法训练 多阶乘计算

蓝桥杯 算法训练 多阶乘计算

资源限制
时间限制：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

思路：
首先计算n(K)!的和，使用一个for循环就OK
算完之后直接算各个位置上的和可以/=10然后再相加

/*
算法训练 多阶乘计算
思路：
	首先计算n(K)!的和，使用一个for循环就OK
	算完之后直接算各个位置上的和可以/=10然后再相加
*/

#include <iostream>
#include <iomanip>
#include <climits>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;
int main()
{
	int n, k, m, sum = 0;
	long long Sum = 0 , temp;
	cin >> n >> k;
	cin >> m;
	for (int i = 1; i <= k; i++)
	{
		temp = 1;
		for (int j = n; j > 0; j -= i) {
			temp *= j;
		}
		Sum += temp;
	}
	if (m == 1)
		cout << Sum;
	else if (m == 2) {
		while (Sum) {
			sum += Sum % 10;
			Sum /= 10;
		}
		cout << sum;
	}
	return 0;
}