結果

問題 No.3202 Periodic Alternating Subsequence
ユーザー suisen
提出日時 2025-07-11 21:56:10
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 412 bytes
コンパイル時間 794 ms
コンパイル使用メモリ 82,192 KB
実行使用メモリ 76,176 KB
最終ジャッジ日時 2025-07-11 21:56:17
合計ジャッジ時間 2,999 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample WA * 2
other WA * 24
権限があれば一括ダウンロードができます

ソースコード

diff #

s=input()
k=int(input())
P=1000000007
T=(((1,1),(0,1)),((1,0),(1,1)))
def m(a,b):
 return(
  ((a[0][0]*b[0][0]+a[0][1]*b[1][0])%P,(a[0][0]*b[0][1]+a[0][1]*b[1][1])%P),
  ((a[1][0]*b[0][0]+a[1][1]*b[1][0])%P,(a[1][0]*b[0][1]+a[1][1]*b[1][1])%P)
 )
X=((1,0),(0,1))
for c in s:
 X=m((T[0]if c=='0'else T[1]),X)
Y=((1,0),(0,1))
while k:
 if k&1:Y=m(Y,X)
 X=m(X,X)
 k=k//2
a=Y[0][0]+Y[0][1]+Y[1][0]+Y[1][1]
print(a%P)
0