結果

問題 No.2899 Taffy Permutation
ユーザー vjudge1vjudge1
提出日時 2024-09-25 20:16:44
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 84 ms / 2,000 ms
コード長 737 bytes
コンパイル時間 1,580 ms
コンパイル使用メモリ 166,088 KB
実行使用メモリ 66,176 KB
最終ジャッジ日時 2024-09-25 20:16:55
合計ジャッジ時間 3,130 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

#define int long long

const int N = 2e3 + 5, mod = 998244353;

int dp[N][N], sum[N][N], n;

char a[N];

signed main() {
  cin >> n;
  for (int i = 1; i <= n; i++) {
    cin >> a[i];
  }
  dp[0][n] = 1;
  for (int i = 1; i <= n; i++) {
    sum[0][i] = 1;
  }
  for (int i = 1; i <= n; i++) {
    for (int j = 0; j <= n; j++) {
      if (a[i] == '0') {
        dp[i][j] = sum[i - 1][j + 1] + dp[i - 1][j] * (n - i + 1 - j) % mod;
        dp[i][j] %= mod;
      }
      else {
        dp[i][j] = (j + 1) * dp[i - 1][j + 1] % mod;
      }
    }
    for (int j = n; j >= 0; j--) {
      sum[i][j] = sum[i][j + 1] + dp[i][j];
      sum[i][j] %= mod;
    }
  }
  cout << dp[n][0];
  return 0;
}
0