結果
問題 | No.2541 Divide 01 String |
ユーザー |
|
提出日時 | 2023-11-24 21:23:00 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 68 ms / 2,000 ms |
コード長 | 511 bytes |
コンパイル時間 | 256 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 65,024 KB |
最終ジャッジ日時 | 2024-09-26 08:48:08 |
合計ジャッジ時間 | 2,452 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
import sys sys.setrecursionlimit(5*10**5) input = sys.stdin.readline from collections import defaultdict, deque, Counter from heapq import heappop, heappush from bisect import bisect_left, bisect_right from math import gcd n = int(input()) s = input().rstrip() dp = [0]*(n+1) dp[0] = 1 last = -1 for i in range(1,n+1): if s[i-1] == "1": dp[i] += dp[i-1] last = i elif last != -1: dp[i] += dp[last-1] dp[i] += dp[i-1] dp[i] %= 998244353 print((dp[-1]-dp[-2])%998244353)