結果

問題 No.658 テトラナッチ数列 Hard
ユーザー 双六双六
提出日時 2020-07-25 15:45:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,843 ms / 2,000 ms
コード長 843 bytes
コンパイル時間 405 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 80,128 KB
最終ジャッジ日時 2024-06-27 07:02:06
合計ジャッジ時間 9,548 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 47 ms
54,400 KB
testcase_01 AC 44 ms
54,144 KB
testcase_02 AC 62 ms
65,024 KB
testcase_03 AC 99 ms
77,024 KB
testcase_04 AC 716 ms
78,848 KB
testcase_05 AC 801 ms
79,104 KB
testcase_06 AC 1,006 ms
79,488 KB
testcase_07 AC 1,060 ms
78,948 KB
testcase_08 AC 1,256 ms
78,896 KB
testcase_09 AC 1,843 ms
79,744 KB
testcase_10 AC 1,839 ms
80,128 KB
権限があれば一括ダウンロードができます

ソースコード

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)
		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