結果

問題 No.2636 No Waiting in Vain
ユーザー tobbie
提出日時 2024-05-07 14:15:50
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 35 ms / 2,000 ms
コード長 1,173 bytes
コンパイル時間 1,593 ms
コンパイル使用メモリ 169,452 KB
実行使用メモリ 6,816 KB
最終ジャッジ日時 2024-11-30 07:40:03
合計ジャッジ時間 3,969 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define MOD 998244353
using ll = long long int;

int mod_div(int x, int y) {
  //return (long long)x * mod_inv(y) % MOD;
  int z = MOD, u = 1, v = 0;
  while (z) {
    int t = y / z;
    y -= t * z; swap(y, z);
    u -= t * v; swap(u, v);
  }
  u %= MOD;
  if (u < 0) u += MOD;
  return (long long)x * u % MOD;
}

int mod_factorial (int x) {
  int ret = 1;
  while (x > 1) {
    ret = ((ll)ret * (ll)x) % MOD;
    x--;
  }
  return ret;
}

int main() {
  int N, K;
  cin >> N >> K;
  string S;
  cin >> S;
  int n = 0, nn = 0;
  rep(i, N) {
    if (S[i] == 'N') {
      if (i == 0)
	n++;
      else if (S[i-1] != 'C')
	n++;
    }
    if (S[i] == 'N')
      nn++;
  }
  // nCr = nPr / r! = n! / (n-r)!*r!
  int nu = mod_factorial(n);
  int ans = 0;
  int fact_nr = nu;
  int fact_r = 1;
  rep(i, nn - K) {
    if (i+1 > n) break;
    fact_nr = mod_div(fact_nr, n-i);
    fact_r = ((ll)fact_r * (ll)(i+1)) % MOD;
    int de = ((ll)fact_nr * (ll)fact_r) % MOD;
    int cnr = mod_div(nu, de) % MOD;
    ans = ((ll)ans + (ll)cnr) % MOD;
  }
  cout << ans << endl;
  return 0;
}
0