結果
問題 |
No.303 割れません
|
ユーザー |
|
提出日時 | 2023-07-04 23:46:46 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,650 ms / 10,000 ms |
コード長 | 703 bytes |
コンパイル時間 | 160 ms |
コンパイル使用メモリ | 81,792 KB |
実行使用メモリ | 183,808 KB |
最終ジャッジ日時 | 2024-07-18 18:43:30 |
合計ジャッジ時間 | 23,928 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 14 |
ソースコード
import sys sys.set_int_max_str_digits(10 ** 9) n = int(input()) if n == 2: print(3) print("INF") exit() print(n) def matpow(A, B, w): l = len(A) while w: if w & 1: C = [0] * l for i in range(l): for j in range(l): C[i] += A[i][j] * B[j] B = C C = [[0] * l for _ in range(l)] for i in range(l): for j in range(l): for k in range(l): C[i][j] += A[i][k] * A[k][j] A = C w >>= 1 return B A = [[1, 1], [1, 0]] B = [1, 0] ans = matpow(A, B, n - 1)[0] if n % 2 == 0: ans -= matpow(A, B, n // 2 - 1)[0] ** 2 print(ans)