結果

問題 No.2326 Factorial to the Power of Factorial to the...
ユーザー AckvyAckvy
提出日時 2023-05-28 15:57:35
言語 Java21
(openjdk 21)
結果
WA  
実行時間 -
コード長 3,024 bytes
コンパイル時間 2,188 ms
コンパイル使用メモリ 77,340 KB
実行使用メモリ 54,328 KB
最終ジャッジ日時 2024-06-08 09:01:16
合計ジャッジ時間 5,534 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 106 ms
40,784 KB
testcase_01 WA -
testcase_02 AC 97 ms
40,016 KB
testcase_03 AC 97 ms
39,940 KB
testcase_04 WA -
testcase_05 AC 106 ms
40,828 KB
testcase_06 AC 106 ms
41,312 KB
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 AC 105 ms
41,160 KB
testcase_11 AC 99 ms
40,040 KB
testcase_12 AC 109 ms
41,244 KB
testcase_13 WA -
testcase_14 AC 131 ms
41,072 KB
testcase_15 WA -
testcase_16 AC 108 ms
40,948 KB
testcase_17 WA -
testcase_18 WA -
testcase_19 AC 99 ms
40,080 KB
testcase_20 AC 108 ms
40,732 KB
testcase_21 AC 108 ms
41,364 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*;

public class Main{
	static final int MOD = 1000000007;

	public static void main(String args[]){
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		int P = sc.nextInt();

		int primes[] = new int[1000];
		int primeNum = 1;
		primes[0] = 2;
		for(int i = 3; i < 1000; i++){
			boolean isPrime = true;
			for(int j = 0; j < primeNum; j++){
				if(i % primes[j] == 0){
					isPrime = false;
					break;
				}
			}
			if(isPrime){
				primes[primeNum] = i;
				primeNum++;
			}
		}

		int dum = P;
		int dividePrime[] = new int[primeNum+1];
		int productPrime[] = new int[primeNum+1];
		for(int i = 0; i < primeNum; i++){
			dividePrime[i] = 0;
			productPrime[i] = 1;
			while(dum % primes[i] == 0){
				dividePrime[i]++;
				productPrime[i] *= primes[i];
				dum /= primes[i];
			}
		}
		if(dum != 1){
			primes[primeNum] = dum;
			dividePrime[primeNum] = 1;
			productPrime[primeNum] = dum;
			primeNum++;
		}

		int maxProdInd = 0;
		for(int i = 0; i < primeNum; i++){
			if(productPrime[maxProdInd] < productPrime[i]){
				maxProdInd = i;
			}
		}

		int maxPrime = primes[maxProdInd];
		int divide = dividePrime[maxProdInd];
//		System.out.println("["+maxPrime+","+divide+"]");

		if(maxPrime > N){
			System.out.println(0);
			return;
		}


		Long numP = 0l;
		for(int i = 1; i <= N; i++){
			int n = i;
			while(n % maxPrime == 0){
				numP++;
				n /= maxPrime;
			}
		}

		long amariP = numP % divide;
		numP /= divide;
//		System.out.println("["+numP+","+amariP+"]");


		long nMDiv = 1;
		long nM = 1;
		long amariNM = 1;
		for(int i = 1; i <= N; i++){
			if(i != divide){
				amariNM *= i;
			}
			amariNM %= MOD;
			nM *= i;
			nM %= MOD;
			nMDiv *= i;
			nMDiv %= divide;
		}
//		System.out.println(nM);

		long divideInv;
		{
			long a = divide;
			long b = MOD;
			long u = 1;
			long v = 0;
			while (b != 0) {
			    long t = a / b;
			    a -= t * b;
			    long dummy = a;
			    a = b;
			    b = dummy;
			    u -= t * v; 
			    dummy = u;
			    u = v;
			    v = dummy;
			}
			u %= MOD;
			if (u < 0) u += MOD;
			divideInv = u;
		}


		char[] S = Long.toBinaryString(nM).toCharArray();
		long binM = nM;
		long ans = 1;

		for(int i = S.length-1; i >= 0; i--){
//			System.out.print(S[i]);
			if(S[i] == '1'){
				ans *= binM;
				ans %= MOD;
			}
			binM *= binM;
			binM %= MOD;
		}

		char[] C = Long.toBinaryString(nMDiv).toCharArray();
		long binMDiv = nMDiv;
		long minusDiv = 1;
		for(int i = C.length-1; i >= 0; i--){
//			System.out.print(S[i]);
			if(C[i] == '1'){
				minusDiv *= binMDiv;
				minusDiv %= divide;
			}
			binMDiv *= binMDiv;
			binMDiv %= divide;
		}
		minusDiv *= amariP;
		minusDiv %= divide;

//		System.out.println();
		long amariAns = ans * amariP;
		amariAns %= MOD;
		amariAns += MOD;
		amariAns -= minusDiv;
		amariAns %= MOD;
		amariAns *= divideInv;
		amariAns %= MOD;
//		System.out.println(amariAns);

		ans *= numP;
		ans %= MOD;
//		System.out.println(ans);
		ans += amariAns;
		ans %= MOD;

		System.out.println(ans);
	}
}
0