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

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

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

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner sca=new Scanner(System.in);
		int n=sca.nextInt();
		int k=sca.nextInt();
		int m=sca.nextInt();
		if(m==1) {
			long sum=0;
			for(int i=1;i<=k;i++) {
				sum+=dfs(n,i);
			}
			System.out.println(sum);
		}else {
			long sum=0;
			for(int i=1;i<=k;i++) {
				sum+=dfs(n,i);
			}
			long sum2=0;
			while(sum/10!=0) {
				sum2+=sum%10;
				sum/=10;
			}
			System.out.println(sum2+sum);
		}
	}
	public static long dfs(int nn,int k) {
		if(nn<1) {
			return 1;
		}	
		return nn*dfs(nn-k,k);
	}

}