結果
問題 | No.93 ペガサス |
ユーザー | maspy |
提出日時 | 2020-04-23 17:18:59 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 469 ms / 5,000 ms |
コード長 | 1,064 bytes |
コンパイル時間 | 147 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 44,532 KB |
最終ジャッジ日時 | 2024-04-21 17:59:14 |
合計ジャッジ時間 | 12,927 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 450 ms
43,888 KB |
testcase_01 | AC | 448 ms
44,148 KB |
testcase_02 | AC | 455 ms
43,768 KB |
testcase_03 | AC | 458 ms
44,532 KB |
testcase_04 | AC | 466 ms
44,404 KB |
testcase_05 | AC | 455 ms
43,896 KB |
testcase_06 | AC | 469 ms
43,992 KB |
testcase_07 | AC | 462 ms
44,020 KB |
testcase_08 | AC | 458 ms
44,152 KB |
testcase_09 | AC | 462 ms
44,408 KB |
testcase_10 | AC | 450 ms
44,388 KB |
testcase_11 | AC | 456 ms
44,152 KB |
testcase_12 | AC | 464 ms
43,892 KB |
testcase_13 | AC | 451 ms
44,276 KB |
testcase_14 | AC | 458 ms
43,900 KB |
testcase_15 | AC | 459 ms
43,892 KB |
testcase_16 | AC | 463 ms
44,400 KB |
testcase_17 | AC | 462 ms
43,888 KB |
testcase_18 | AC | 459 ms
44,156 KB |
testcase_19 | AC | 463 ms
43,892 KB |
ソースコード
import sys read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines import numpy as np MOD = 10 ** 9 + 7 N = int(read()) if N == 1: print(1) exit() M = N // 2 def update_dp(f, g): newf = np.zeros(len(f) + 1, np.int64) newg = newf.copy() newg[:-1] = f + f + g newf[1:] = f + g newf %= MOD newg %= MOD return newf, newg f = np.array([0, 1], np.int64) g = np.array([0, 0], np.int64) for _ in range((N - 2) // 2): f, g = update_dp(f, g) A = f + g if N & 1: f, g = update_dp(f, g) B = f + g def convolve(A, B): A1, A2 = divmod(A, 1 << 16) B1, B2 = divmod(B, 1 << 16) X = np.convolve(A1, B1) % MOD Z = np.convolve(A2, B2) % MOD Y = (np.convolve(A1 + A2, B1 + B2) - X - Z) % MOD return ((X << 32) + (Y << 16) + Z) % MOD A = convolve(A, B) fact = [1, 1] for n in range(2, N + 10): fact.append(fact[-1] * n % MOD) fact = np.array(fact, dtype=np.int64) A *= fact[:N+1] A[1 - (N & 1)::2] *= - 1 A %= MOD answer = A.sum() % MOD print(answer)