結果

問題 No.658 テトラナッチ数列 Hard
ユーザー 双六双六
提出日時 2020-07-25 15:46:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,885 ms / 2,000 ms
コード長 924 bytes
コンパイル時間 196 ms
コンパイル使用メモリ 82,376 KB
実行使用メモリ 80,724 KB
最終ジャッジ日時 2024-06-27 07:03:38
合計ジャッジ時間 10,110 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
54,272 KB
testcase_01 AC 40 ms
54,656 KB
testcase_02 AC 54 ms
67,304 KB
testcase_03 AC 84 ms
76,672 KB
testcase_04 AC 738 ms
78,848 KB
testcase_05 AC 819 ms
79,004 KB
testcase_06 AC 1,020 ms
79,020 KB
testcase_07 AC 1,128 ms
79,872 KB
testcase_08 AC 1,297 ms
79,336 KB
testcase_09 AC 1,882 ms
79,440 KB
testcase_10 AC 1,885 ms
80,724 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)
		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