結果
問題 | No.2541 Divide 01 String |
ユーザー |
|
提出日時 | 2023-11-24 22:58:35 |
言語 | D (dmd 2.109.1) |
結果 |
AC
|
実行時間 | 5 ms / 2,000 ms |
コード長 | 1,813 bytes |
コンパイル時間 | 3,899 ms |
コンパイル使用メモリ | 170,368 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-09-26 09:44:54 |
合計ジャッジ時間 | 3,821 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
import std; void main () { int N = readln.chomp.to!int; string S = readln.chomp; solve(N, S); } void solve (int N, string S) { /* dp的に解く 新しい区間を導入したとき、そこで切るか切らないかの2択 */ bool ok = false; foreach (c; S) if (c == '1') ok = true; if (!ok) { writeln(0); return; } const int MOD = 998244353; long ans = 1; // カットしないやつ // 右と左の0を削除 foreach (i, c; S) if (c == '1') { S = S[i..$]; break; } foreach_reverse (i, c; S) if (c == '1') { S = S[0..i+1]; break; } int len = 0; int cur = 1; while (true) { if (cur == S.length) { break; } if (S[cur] == '1') { ans *= len+2; ans %= MOD; len = -1; } len++; cur++; } writeln(ans); } long modPow (long a, long x, ref const long MOD) { import std.exception : enforce; enforce(0 <= x, "x must satisfy 0 <= x"); enforce(1 <= MOD, "MOD must satisfy 1 <= MOD"); enforce(MOD <= int.max, "MOD must satisfy MOD*MOD <= long.max"); // normalize a %= MOD; a += MOD; a %= MOD; long res = 1L; long base = a; while (0 < x) { if (0 < (x&1)) (res *= base) %= MOD; (base *= base) %= MOD; x >>= 1; } return res % MOD; } T modPow (T, X, Y) (T a, X x, ref const Y MOD) { import std.exception: enforce; enforce(0 <= x, "x must satisfy 0 <= x"); enforce(1 <= MOD, "MOD must satisfy 1 <= MOD"); // normalize a %= MOD; a += MOD; a %= MOD; T res = 1; T base = a; while (0 < x) { if (0 < (x&1)) (res *= base) %= MOD; (base *= base) %= MOD; x >>= 1; } return res % MOD; }