結果

問題 No.1504 ヌメロニム
ユーザー そすうぽよそすうぽよ
提出日時 2020-10-18 04:59:25
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 2,246 bytes
コンパイル時間 1,072 ms
コンパイル使用メモリ 96,044 KB
実行使用メモリ 17,408 KB
最終ジャッジ日時 2024-09-20 06:09:47
合計ジャッジ時間 12,053 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 AC 11 ms
12,800 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 10 ms
12,672 KB
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 AC 11 ms
12,800 KB
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 AC 11 ms
12,732 KB
testcase_20 RE -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 RE -
testcase_38 RE -
testcase_39 RE -
testcase_40 RE -
testcase_41 RE -
testcase_42 RE -
testcase_43 RE -
testcase_44 RE -
testcase_45 RE -
testcase_46 RE -
testcase_47 RE -
testcase_48 RE -
testcase_49 RE -
testcase_50 WA -
testcase_51 WA -
testcase_52 WA -
testcase_53 WA -
testcase_54 WA -
testcase_55 RE -
testcase_56 WA -
testcase_57 AC 11 ms
12,928 KB
testcase_58 WA -
testcase_59 WA -
testcase_60 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cmath>
#include <cstdint>
#include <complex>
#include <iostream>
#include <vector>
#define FOR(i,k,n)  for(int i = (k);i < (n);++i)
#define REP(i,n)    FOR(i,0,n)
#define ALL(x)      begin(x),end(x)

using namespace std;
using vecint = vector<int>;
using ll = int64_t;
using vecll = vector<ll>;

using P = complex<double>;

const double pi = acos(-1.0);

vector<P> FFT(double theta, const vector<P> &a) {
  const int n = a.size();
  vector<P> ret = a;
  for (int m = n; m >= 2; m >>= 1, theta *= 2) {
    REP(i,m/2) {
      for (int j = i; j < n; j += m) {
        int k = j + m / 2;
        P x = ret[j] - ret[k];
        ret[j] += ret[k];
        ret[k] = exp(i * theta * P(0, 1)) * x;
      }
    }
  }
  for (int i = 0, j = 1; j < n - 1; j++) {
    for (int k = n >> 1; k > (i ^= k); k >>= 1) {;}
    if (j < i) swap(ret[i], ret[j]);
  }
  return ret;
}

vector<ll> convolution(const vector<ll> &lhs, const vector<ll> &rhs) {
  int n = 1, a = lhs.size(), b = rhs.size();
  while (n < max(a, b) * 2) n <<= 1;
  vector<P> temp1(n), temp2(n);
  REP(i,n/2) {
    if (i < a) temp1[i] = P(lhs[i], 0);
    if (i < b) temp2[i] = P(rhs[i], 0);
  }
  temp1 = FFT(2.0 * pi / n, temp1);
  temp2 = FFT(2.0 * pi / n, temp2);
  REP(i,n) temp1[i] *= temp2[i];
  temp1 = FFT(-2.0 * pi / n, temp1);
  vector<ll> ret(n);
  REP(i,n) ret[i] = temp1[i].real() / n + 0.5;
  return ret;
}

constexpr ll MOD = 998244353;
constexpr ll MAX_F = 600000;

// a^-1 mod p
ll inv(ll a,ll p){
  return ( a == 1 ? 1 : (1 - p*inv(p%a,a)) / a + p );
}

int main() {
  vecll fact(MAX_F+1), finv(MAX_F+1);
  fact[0] = 1;
  REP(i,MAX_F) {
    fact[i+1] = fact[i] * (i+1) % MOD;
  }
  finv[MAX_F] = inv(fact[MAX_F], MOD);
  REP(ri,MAX_F) {
    int i = MAX_F - ri;
    finv[i-1] = finv[i] * i % MOD;
  }
  auto comb = [&] (ll n, ll k) {
    return fact[n] * finv[k] % MOD * finv[n-k] % MOD;
  };
  ll n,k;
  cin>>n>>k;
  string s;
  cin>>s;
  vecll left_i(n, 0), right_n(n, 0);
  REP(i,n) {
    if (s[i] == 'i') {
      left_i[i] = 1;
    } else {
      right_n[n-1-i] = 1;
    }
  }
  auto poly = convolution(left_i, right_n);
  ll ans = 0;
  REP(i,n-k-1) {
    ll span = n-2-i;
    ans += poly[i] * comb(span,k) % MOD;
    ans %= MOD;
  }
  cout<<ans<<"\n";
  return 0;
}
0