結果

問題 No.117 組み合わせの数
ユーザー t8m8⛄️t8m8⛄️
提出日時 2016-02-27 02:11:48
言語 Java21
(openjdk 21)
結果
AC  
実行時間 1,071 ms / 5,000 ms
コード長 2,582 bytes
コンパイル時間 4,075 ms
コンパイル使用メモリ 85,068 KB
実行使用メモリ 106,976 KB
最終ジャッジ日時 2024-09-24 10:59:59
合計ジャッジ時間 6,650 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,071 ms
106,976 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*;
import java.io.*;
import java.awt.geom.*;
import java.math.*;

public class No0117 {

	static final Scanner in = new Scanner(System.in);
	static final PrintWriter out = new PrintWriter(System.out,false);
	static boolean debug = false;

	static final long MOD = 1_000_000_007;

	static long[] extgcd(long a, long b) {
		if (a == 0 || b == 0) return null;
        int asig = Long.signum(a);
        int bsig = Long.signum(b);
        a = Math.abs(a); b = Math.abs(b);
        long x = 1, y = 0;
        long p = 0, q = 1;
        while (b > 0) {
            long c = a/b, d = a;
            a = b; b = d % b;
            d = x; x = p; p = d - c*p;
            d = y; y = q; q = d - c*q;
        }
        return new long[]{x*asig,y*bsig,a};
	}

	static long inv(long a, long mod) {
		long[] gcd = extgcd(a, mod);
		return (mod + gcd[0]%mod)%mod;
	}

	static long[] fact(int n, long mod) {
		long[] fact = new long[n+1];
		fact[0] = fact[1] = 1;
		for (int i=2; i<=n; i++) {
			fact[i] = fact[i-1]*i%mod;
		}
		return fact;
	}

	static long[] factInv(long[] fact, long mod) {
		int n = fact.length;
		long[] inv = new long[n];
		inv[n-1] = inv(fact[n-1], mod);
		for (int i=n-1; i>0; i--) {
			inv[i-1] = inv[i]*i%mod;
		}
		return inv;
	}

	static long nPk(int n, int k, long mod, long[] fact, long[] factI) {
		if (k < 0 || n < k) return 0;
		return fact[n] * factI[n-k] % mod;
	}

	static long nCk(int n, int k, long mod, long[] fact, long[] factI) {
		if (k < 0 || n < k) return 0;
		return fact[n] * factI[k] % mod * factI[n-k] % mod;
	}

	static long nHk(int n, int k, long mod, long[] fact, long[] factI) {
		if (n == 0 && k == 0) return 1;
		return nCk(n+k-1, k, mod, fact, factI);
	}

	static void solve() {
		long[] fact = fact(2000000, MOD);
		long[] factI = factInv(fact, MOD);

		int t = in.nextInt();

		while (t-- > 0) {
			String[] s = in.next().split("[\\(,\\)]");
			if ("C".equals(s[0])) {
				out.println(nCk(Integer.parseInt(s[1]), Integer.parseInt(s[2]), MOD, fact, factI));
			} else if ("P".equals(s[0])) {
				out.println(nPk(Integer.parseInt(s[1]), Integer.parseInt(s[2]), MOD, fact, factI));
			} else {
				out.println(nHk(Integer.parseInt(s[1]), Integer.parseInt(s[2]), MOD, fact, factI));
			}
		}
	}

	public static void main(String[] args) {
		debug = args.length > 0;
		long start = System.currentTimeMillis();

		solve();
		out.flush();

		long end = System.currentTimeMillis();
		dump((end-start) + "ms");
		in.close();
		out.close();
	}

	static void dump(Object... o) { if (debug) System.err.println(Arrays.deepToString(o)); }
}
0