結果
問題 | No.2989 Fibonacci Prize |
ユーザー |
![]() |
提出日時 | 2024-12-14 11:38:26 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 902 bytes |
コンパイル時間 | 465 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 138,496 KB |
最終ジャッジ日時 | 2024-12-14 11:38:35 |
合計ジャッジ時間 | 7,348 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 WA * 1 |
other | AC * 65 WA * 12 |
ソースコード
N, M = map(int, input().split())def solve(N, M):if N == 1:ans = Mif M >= 3:ans = (M - 2) * (M - 1) // 2 + 3return ansif M <= 2:return 0fib = [0, 1]for i in range(2, M + 1):fib.append((fib[-1] + fib[-2]) % N)if fib[0] == fib[-2] and fib[1] == fib[-1]:breakFib = fib[:]Fib.pop()Fib.pop()n = len(Fib)R = [0 for i in range(N)]for i in range(n - 1):Fib[i + 1] = (Fib[i + 1] + Fib[i]) % Nfor r in Fib:R[r] += 1x = (M - 1) // nans = 0# R <= M - 2, L <= Rのとき条件満たす.for i in range(N):R[i] *= xr = R[i]ans += r * (r - 1) // 2for i in range(M - 1 - x * n):r = Fib[i]ans += R[r]R[r] += 1# fib_M = fib_(M - 1) + fib_(M - 2)if fib[M - x * n] % N == 0:ans += 2# fib_(M - 1)if fib[M - x * n - 1] % N == 0:ans += 1return ansprint(solve(N, M))