結果

問題 No.2837 Flip Triomino
ユーザー ripity
提出日時 2024-08-09 22:29:09
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 8 ms / 2,000 ms
コード長 1,075 bytes
コンパイル時間 2,443 ms
コンパイル使用メモリ 205,440 KB
最終ジャッジ日時 2025-02-23 21:47:29
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

#include <atcoder/modint>
using mint = atcoder::modint998244353;

int main() {
  int N, M;
  cin >> N >> M;
  vector<string> S(N);
  for(int i = 0; i < N; i++) {
    cin >> S[i];
  }
  vector<int> cnt_q(N + M - 1);
  vector<int> cnt_b(N + M - 1);
  for(int i = 0; i < N; i++) {
    for(int j = 0; j < M; j++) {
      cnt_q[i + j] += (S[i][j] == '?');
      cnt_b[i + j] += (S[i][j] == 'B');
    }
  }
  vector dp(N + M - 1, vector<mint>(4));
  for(int f0 : {0, 1}) for(int f1 : {0, 1}) {
    if(cnt_q[0] == 0 && f0 != cnt_b[0] % 2) continue;
    if(cnt_q[1] == 0 && f1 != cnt_b[1] % 2) continue;
    dp[1][(f0 << 0) | (f1 << 1)] = mint(2).pow(max(0, cnt_q[0] - 1) + max(0, cnt_q[1] - 1));
  }
  for(int i = 2; i < N + M - 1; i++) {
    for(int f0 : {0, 1}) for(int f1 : {0, 1}) for(int f2 : {0, 1}) {
      if(cnt_q[i] == 0 && f2 != cnt_b[i] % 2) continue;
      dp[i][((f1 ^ f0) << 0) | ((f2 ^ f0) << 1)] += dp[i - 1][(f0 << 0) | (f1 << 1)] * mint(2).pow(max(0, cnt_q[i] - 1));
    }
  }
  cout << dp.back()[0].val() << endl;
}
0