結果
問題 | No.978 Fibonacci Convolution Easy |
ユーザー | yakamoto |
提出日時 | 2020-02-02 23:22:39 |
言語 | Kotlin (1.9.23) |
結果 |
RE
|
実行時間 | - |
コード長 | 2,373 bytes |
コンパイル時間 | 17,808 ms |
コンパイル使用メモリ | 457,040 KB |
実行使用メモリ | 85,548 KB |
最終ジャッジ日時 | 2023-10-19 01:32:11 |
合計ジャッジ時間 | 26,879 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 308 ms
53,932 KB |
testcase_01 | AC | 352 ms
62,820 KB |
testcase_02 | AC | 338 ms
56,684 KB |
testcase_03 | AC | 379 ms
85,532 KB |
testcase_04 | AC | 342 ms
56,680 KB |
testcase_05 | AC | 326 ms
53,956 KB |
testcase_06 | AC | 350 ms
60,788 KB |
testcase_07 | AC | 359 ms
70,380 KB |
testcase_08 | AC | 349 ms
62,824 KB |
testcase_09 | AC | 363 ms
77,300 KB |
testcase_10 | AC | 376 ms
85,548 KB |
testcase_11 | AC | 344 ms
58,740 KB |
testcase_12 | AC | 319 ms
53,944 KB |
testcase_13 | AC | 341 ms
60,772 KB |
testcase_14 | AC | 335 ms
56,612 KB |
testcase_15 | AC | 346 ms
60,776 KB |
testcase_16 | AC | 378 ms
85,548 KB |
testcase_17 | AC | 377 ms
85,528 KB |
testcase_18 | RE | - |
testcase_19 | AC | 298 ms
51,876 KB |
testcase_20 | AC | 294 ms
51,880 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 Solver(stream: InputStream, private val out: java.io.PrintWriter) { fun solve() { val N = ni() val p = ni() val A = LongArray(N) A[0] = 0 A[1] = 1 for (i in 2 until N) { A[i] = (A[i - 1] * p % MOD + A[i - 2]) % MOD } val cum = LongArray(N + 1) for (i in 0 until N) { cum[i + 1] = (cum[i] + A[i]) % MOD } var ans = 0L for (i in 0 until N) { ans += A[i] * cum[i + 1] % MOD } 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, offset) { ni() } } private inline fun map(n: Int, offset: Int = 0, f: (Int) -> Int): IntArray { val res = IntArray(n) for (i in 0 until n) { res[i] = f(i + offset) } 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) } } } /** * 勝手にimport消されるのを防ぎたい */ private fun hoge() { min(1, 2) max(1, 2) abs(-10) } } fun <A, B> pair(a: A, b: B) = RPair(a, b) data class RPair<A, B>(val _1: A, val _2: B) fun main() { val out = java.io.PrintWriter(System.out) Solver(System.`in`, out).solve() out.flush() }