結果

問題 No.503 配列コレクション
ユーザー 37zigen37zigen
提出日時 2017-04-08 03:15:05
言語 Java21
(openjdk 21)
結果
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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 309 ms
101,688 KB
testcase_01 AC 302 ms
101,536 KB
testcase_02 AC 301 ms
101,664 KB
testcase_03 AC 304 ms
101,788 KB
testcase_04 AC 294 ms
101,584 KB
testcase_05 AC 304 ms
101,808 KB
testcase_06 AC 298 ms
101,508 KB
testcase_07 AC 298 ms
101,808 KB
testcase_08 AC 305 ms
101,532 KB
testcase_09 AC 312 ms
101,616 KB
testcase_10 AC 304 ms
101,668 KB
testcase_11 AC 305 ms
101,600 KB
testcase_12 AC 305 ms
101,800 KB
testcase_13 AC 304 ms
101,856 KB
testcase_14 AC 306 ms
101,748 KB
testcase_15 AC 305 ms
101,872 KB
testcase_16 AC 306 ms
101,820 KB
testcase_17 AC 299 ms
101,804 KB
testcase_18 AC 318 ms
101,680 KB
testcase_19 AC 311 ms
101,616 KB
testcase_20 AC 302 ms
101,892 KB
testcase_21 AC 300 ms
101,492 KB
testcase_22 AC 297 ms
101,952 KB
testcase_23 AC 304 ms
101,844 KB
testcase_24 AC 315 ms
102,032 KB
権限があれば一括ダウンロードができます

ソースコード

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