結果

問題 No.503 配列コレクション
ユーザー 37zigen37zigen
提出日時 2017-04-08 03:10:30
言語 Java21
(openjdk 21)
結果
RE  
実行時間 -
コード長 1,423 bytes
コンパイル時間 2,645 ms
コンパイル使用メモリ 77,640 KB
実行使用メモリ 101,892 KB
最終ジャッジ日時 2024-07-16 04:25:05
合計ジャッジ時間 11,721 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 293 ms
101,680 KB
testcase_01 AC 316 ms
101,744 KB
testcase_02 AC 313 ms
101,852 KB
testcase_03 AC 291 ms
101,772 KB
testcase_04 RE -
testcase_05 AC 293 ms
101,804 KB
testcase_06 AC 296 ms
101,812 KB
testcase_07 AC 303 ms
101,668 KB
testcase_08 AC 296 ms
101,828 KB
testcase_09 AC 302 ms
101,664 KB
testcase_10 AC 296 ms
101,844 KB
testcase_11 AC 303 ms
101,528 KB
testcase_12 AC 316 ms
101,560 KB
testcase_13 AC 323 ms
101,800 KB
testcase_14 AC 305 ms
101,688 KB
testcase_15 AC 300 ms
101,664 KB
testcase_16 AC 294 ms
101,892 KB
testcase_17 AC 293 ms
101,868 KB
testcase_18 AC 309 ms
101,664 KB
testcase_19 RE -
testcase_20 AC 288 ms
101,764 KB
testcase_21 RE -
testcase_22 AC 307 ms
101,856 KB
testcase_23 AC 294 ms
101,676 KB
testcase_24 RE -
権限があれば一括ダウンロードができます

ソースコード

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)
		return fac[a + b - 1] * invfac[a] % mo * invfac[b - 1] % mo;
	}

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