結果
問題 |
No.3202 Periodic Alternating Subsequence
|
ユーザー |
|
提出日時 | 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 |
ソースコード
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)