結果
問題 | No.1102 Remnants |
ユーザー | yakamoto |
提出日時 | 2020-07-03 23:09:28 |
言語 | Kotlin (1.9.23) |
結果 |
AC
|
実行時間 | 523 ms / 2,000 ms |
コード長 | 3,744 bytes |
コンパイル時間 | 18,143 ms |
コンパイル使用メモリ | 454,700 KB |
実行使用メモリ | 64,452 KB |
最終ジャッジ日時 | 2023-10-17 06:11:33 |
合計ジャッジ時間 | 30,684 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 285 ms
51,848 KB |
testcase_01 | AC | 286 ms
51,848 KB |
testcase_02 | AC | 288 ms
51,856 KB |
testcase_03 | AC | 290 ms
51,840 KB |
testcase_04 | AC | 289 ms
51,852 KB |
testcase_05 | AC | 496 ms
58,136 KB |
testcase_06 | AC | 430 ms
57,544 KB |
testcase_07 | AC | 508 ms
59,828 KB |
testcase_08 | AC | 304 ms
51,920 KB |
testcase_09 | AC | 367 ms
54,456 KB |
testcase_10 | AC | 296 ms
51,908 KB |
testcase_11 | AC | 364 ms
54,444 KB |
testcase_12 | AC | 368 ms
54,460 KB |
testcase_13 | AC | 369 ms
54,444 KB |
testcase_14 | AC | 467 ms
57,640 KB |
testcase_15 | AC | 458 ms
57,624 KB |
testcase_16 | AC | 523 ms
62,132 KB |
testcase_17 | AC | 520 ms
58,060 KB |
testcase_18 | AC | 523 ms
64,452 KB |
testcase_19 | AC | 452 ms
57,632 KB |
testcase_20 | AC | 420 ms
54,756 KB |
testcase_21 | AC | 480 ms
59,852 KB |
testcase_22 | AC | 511 ms
60,392 KB |
testcase_23 | AC | 475 ms
59,780 KB |
testcase_24 | AC | 470 ms
57,644 KB |
testcase_25 | AC | 504 ms
63,888 KB |
testcase_26 | AC | 373 ms
54,488 KB |
testcase_27 | AC | 450 ms
57,640 KB |
ソースコード
import java.io.BufferedReader import java.io.InputStream import java.io.InputStreamReader import java.util.* import kotlin.math.abs import kotlin.math.max import kotlin.math.min val MOD = 1_000_000_007 class Comb(n: Int, val mod: Int) { val F = LongArray(n + 1) val I = LongArray(n + 1) init { F[0] = 1 for (i in 1..n) { F[i] = F[i - 1] * i % mod } I[n] = powMod(F[n], (mod - 2).toLong(), mod) for (i in n - 1 downTo 0) { I[i] = I[i + 1] * (i + 1) % mod } } private fun powMod(a: Long, n: Long, mod: Int): Long { if (n == 0L) return 1 val res = powMod(a * a % mod, n / 2, mod) return if (n % 2 == 1L) res * a % mod else res } fun comb(n: Int, k: Int): Long { return F[n] * I[k] % mod * I[n - k] % mod } fun perm(n: Int, k: Int): Long { return F[n] * I[n - k] % mod } fun inv(x: Int): Long { return I[x] * F[x - 1] % mod } /** * nのグループからk回重複ありで選ぶ組み合わせ数 * n - 1のしきりとkの○で考える */ fun H(n: Int, k: Int) = comb(n + k - 1, k) } class Solver(stream: InputStream, private val out: java.io.PrintWriter) { fun solve() { val N = ni() val K = ni() val A = na(N) var ans = 0L val dp = LongArray(N + 1) dp[0] = 1 val comb = Comb(N, MOD) for (i in 1 .. N) { dp[i] = dp[i - 1] * (K + i) % MOD * comb.inv(i) % MOD } debug(dp) fun c(i: Int): Long { return dp[i - 1] // return comb.comb(i + K - 1, K) } for (i in 0 until N) { ans += c(N - i) * c(i + 1) % MOD * A[i] % MOD } out.println(ans % MOD) } private val isDebug = try { // なんか本番でエラーでる System.getenv("MY_DEBUG") != null } catch (t: Throwable) { false } private var tokenizer: StringTokenizer? = null private val reader = BufferedReader(InputStreamReader(stream), 32768) private fun next(): String { while (tokenizer == null || !tokenizer!!.hasMoreTokens()) { tokenizer = StringTokenizer(reader.readLine()) } return tokenizer!!.nextToken() } private fun ni() = next().toInt() private fun nl() = next().toLong() private fun ns() = next() private fun na(n: Int, offset: Int = 0): IntArray { return map(n) { ni() + offset } } private fun nal(n: Int, offset: Int = 0): LongArray { val res = LongArray(n) for (i in 0 until n) { res[i] = nl() + offset } return res } private fun na2(n: Int, offset: Int = 0): Array<IntArray> { val a = Array(2){IntArray(n)} for (i in 0 until n) { for (e in a) { e[i] = ni() + offset } } return a } private inline fun map(n: Int, f: (Int) -> Int): IntArray { val res = IntArray(n) for (i in 0 until n) { res[i] = f(i) } return res } private inline fun debug(msg: () -> String) { if (isDebug) System.err.println(msg()) } private fun debug(a: LongArray) { debug { a.joinToString(" ") } } private fun debug(a: IntArray) { debug { a.joinToString(" ") } } private fun debug(a: BooleanArray) { debug { a.map { if (it) 1 else 0 }.joinToString("") } } private fun debugDim(A: Array<LongArray>) { if (isDebug) { for (a in A) { debug(a) } } } private fun debugDim(A: Array<IntArray>) { if (isDebug) { for (a in A) { debug(a) } } } /** * 勝手にimport消されるのを防ぎたい */ private fun hoge() { min(1, 2) max(1, 2) abs(-10) } } fun main() { val out = java.io.PrintWriter(System.out) Solver(System.`in`, out).solve() out.flush() }