結果

問題 No.3037 Restricted Lucas (Hard)
ユーザー googol_S0googol_S0
提出日時 2020-09-24 15:12:31
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 931 bytes
コンパイル時間 842 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 70,016 KB
最終ジャッジ日時 2024-06-28 04:55:18
合計ジャッジ時間 1,617 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

zero=sum([False,False])
one=sum([True,zero])
two=sum([one,one])
three=sum([one,two])
four=sum([one,three])
five=sum([one,four])
six=sum([one,five])
seven=sum([one,six])
eight=sum([one,seven])
nine=sum([one,eight])
ten=sum([one,nine])
def mul(a,b):
  r=zero
  while b:
    if b&one:
      r=sum([r,a])
    a<<=one
    b>>=one
  return r

mod=int(pow(ten,nine)+seven)
def m(a,b):
  r=[[zero]*len(b[zero]) 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])):
        r[i][j]=divmod(sum([r[i][j],a[i][k]*b[k][j]]),mod)[one]
  return r

def p(a,n):
  r=[[zero]*len(a) for i in range(len(a))]
  b=[]
  for i in range(len(a)):
    r[i][i]=one
    b.append(a[i][:])
  l=n
  while l>zero:
    if l&one:
      r=m(b,r)
    b=m(b,b)
    l>>=one
  return r

B=[[zero,two],[zero,one]]
A=[[zero,one],[one,one]]
T=int(input())
for i in range(T):
  print(m(p(A,int(input())),B)[zero][one])
0