結果

問題 No.3500 01 String
コンテスト
ユーザー titia
提出日時 2026-04-19 05:27:45
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 122 ms / 2,000 ms
コード長 809 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 177 ms
コンパイル使用メモリ 85,504 KB
実行使用メモリ 137,728 KB
最終ジャッジ日時 2026-04-19 05:27:58
合計ジャッジ時間 3,724 ms
ジャッジサーバーID
（参考情報） 		judge1_1 / judge3_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
input = sys.stdin.readline

mod=998244353

N=int(input())
A=list(map(int,list(input().strip())))

while A and A[-1]==0:
    A.pop()

start=-1
for i in range(len(A)):
    if A[i]==0:
        start=i
        break

if start==-1:
    print(1)
else:
    A=A[start:]

    DP1=[0]*len(A)# 0が連続している
    DP2=[0]*len(A)# 0終了。1が入る。次の10までそのまま

    for i in range(len(A)):
        if i==0:
            DP1[i]=1
            DP2[i]=1
            
        elif (A[i-1]==1 and A[i]==0):
            DP1[i]=DP1[i-1]+DP2[i-1]
            DP2[i]=DP2[i-1]+DP1[i-1]
            

        else:
            DP2[i]=DP2[i-1]+DP1[i-1]
            DP1[i]=DP1[i-1]

        DP1[i]%=mod
        DP2[i]%=mod

    print((DP1[-1]+DP2[-1])%mod)
0