結果

問題 No.658 テトラナッチ数列 Hard
ユーザー 双六双六
提出日時 2020-07-25 15:46:58
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 924 bytes
コンパイル時間 1,253 ms
コンパイル使用メモリ 87,244 KB
実行使用メモリ 81,724 KB
最終ジャッジ日時 2023-09-09 14:08:49
合計ジャッジ時間 12,092 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 94 ms
71,248 KB
testcase_01 AC 94 ms
71,472 KB
testcase_02 AC 112 ms
77,384 KB
testcase_03 AC 134 ms
78,596 KB
testcase_04 AC 792 ms
81,544 KB
testcase_05 AC 896 ms
81,076 KB
testcase_06 AC 1,104 ms
80,864 KB
testcase_07 AC 1,202 ms
80,572 KB
testcase_08 AC 1,376 ms
80,720 KB
testcase_09 AC 1,999 ms
81,724 KB
testcase_10 TLE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys; input = sys.stdin.buffer.readline
sys.setrecursionlimit(10**7)
from collections import defaultdict

def getlist():
	return list(map(int, input().split()))

mod = 17

def mul(A, B):
	C = [[0] * len(B[0]) for i in range(len(A))]
	for i in range(len(A)):
		for k in range(len(B)):
			for j in range(len(B[0])):
				C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % mod
	return C

def matrixPow(A, n, N):
	B = [[0] * N for i in range(N)]
	for i in range(N):
		B[i][i] = 1

	while n > 0:
		if n & 1 == 1:
			B = mul(A, B)
		A = mul(A, A)
		for i in range(N):
			for j in range(N):
				A[i][j] %= mod
				B[i][j] %= mod
		n = n >> 1

	return B

#処理内容
def main():
	Q = int(input())
	A = [[1, 1, 1, 1], [1, 0, 0, 0], [0, 1, 0, 0], [0, 0, 1, 0]]

	for i in range(Q):
		n = int(input())
		B = matrixPow(A, n - 1, 4)
		s = mul(B, [[1], [0], [0], [0]])
		ans = s[3][0] % 17
		print(ans)



if __name__ == '__main__':
	main()
0