結果

問題 No.2222 Respawn
コンテスト
ユーザー stoq
提出日時 2022-05-31 03:32:46
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 901 bytes
コンパイル時間 2,032 ms
コンパイル使用メモリ 202,204 KB
最終ジャッジ日時 2025-01-29 17:10:58
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 2 WA * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/modint>
using mint = atcoder::modint998244353;

// f(x) = a + bx
using linear = pair<mint, mint>;
linear operator+(linear f, linear g) {
  return {f.first + g.first, f.second + g.second};
}
linear operator+(linear f, mint a) { return {f.first + a, f.second}; }
linear operator*(linear f, mint a) { return {f.first * a, f.second * a}; }
// f(x) = 0
mint solve(linear f) { return -f.first / f.second; }

int main() {
  int n;
  string s;
  cin >> n >> s;
  vector<mint> E(n + 1);
  vector<linear> f(n + 1);
  f[n - 1] = {0, 0};
  f[n] = {0, 1};
  const mint inv3 = mint(3).inv();

  for (int i = n - 2; i >= 0; i--) {
    if (s[i] == '#') {
      f[i] = {0, 1};
      continue;
    }
    E[i] = solve((f[i + 1] + f[i + 2] + linear{0, -2}) * inv3 + 1);
    f[i] = (f[i + 1] + f[i + 2] + E[i]) * inv3 + 1;
  }
  cout << E[0].val() << "\n";
}
0