結果

問題 No.503 配列コレクション
ユーザー 37zigen
提出日時 2017-04-08 03:15:05
言語 Java
(openjdk 23)
結果
AC  
実行時間 318 ms / 2,000 ms
コード長 1,496 bytes
コンパイル時間 2,304 ms
コンパイル使用メモリ 77,988 KB
実行使用メモリ 102,032 KB
最終ジャッジ日時 2024-07-16 04:26:58
合計ジャッジ時間 11,035 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.ArrayDeque;
import java.util.Arrays;
import java.util.Scanner;

class Main {
	public static void main(String[] args) {
		new Main().run();
	}

	final long mo = 1_000_000_000 + 7;

	void run() {
		long[] fac = new long[2_000_000];
		long[] invfac = new long[fac.length];
		long[] inv = new long[fac.length];
		fac[0] = 1;
		inv[0] = 1;
		inv[1] = 1;
		invfac[0] = 1;
		for (int i = 1; i < fac.length; ++i) {
			fac[i] = fac[i - 1] * i % mo;
		}
		for (int i = 2; i < inv.length; ++i) {
			// 0=p/i*i+p%i
			// i^(-1)=-1*(p/i)*(p%i)^(-1)
			inv[i] = (mo - inv[(int) (mo % i)] * (mo / i) % mo) % mo;
		}
		for (int i = 1; i < invfac.length; ++i) {
			invfac[i] = invfac[i - 1] * inv[i] % mo;
		}
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		int K = sc.nextInt();
		int D = sc.nextInt();
		int q = N / (K - 1);
		if (N % (K - 1) == 0)
			--q;
		int res = N - q * (K - 1);
		if (D == 1) {
			System.out.println(res);
			return;
		}
		long comb = 0;
		long pow = 1;
		for (int i = 0; i <= q; ++i) {
			comb = comb + pow * H(q - i, res - 1, fac, invfac, mo);
			comb %= mo;
			pow = (pow * D) % mo;
		}
		comb = (comb * res) % mo;
		System.out.println(comb);
	}

	long H(int a, int b, long[] fac, long[] invfac, long mo) {
		// (a+b-1)C(a)
		if (a > 0 && b == 0)
			return 0;
		if (a == 0 && b == 0)
			return 1;
		return fac[a + b - 1] * invfac[a] % mo * invfac[b - 1] % mo;
	}

	void tr(Object... objects) {
		System.out.println(Arrays.deepToString(objects));
	}
}
0