結果

問題 No.1504 ヌメロニム
コンテスト
ユーザー そすうぽよ
提出日時 2020-12-12 07:30:16
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 3,220 bytes
コンパイル時間 1,181 ms
コンパイル使用メモリ 101,228 KB
最終ジャッジ日時 2025-01-16 22:47:57
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 38 TLE * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

// bit並列+NTTで畳み込む ver
#include <cmath>
#include <cstdint>
#include <bitset>
#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>;

const double pi = acos(-1.0);

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

struct Mod {
  int n;
  Mod () : n(0) {;}
  Mod (int m) : n(m) {
    if (n >= MOD) n %= MOD;
    else if (n < 0) n = (n % MOD + MOD) % MOD;
  }
  operator int() { return n; }
};

bool operator==(Mod a, Mod b) { return a.n == b.n; }
Mod operator+=(Mod &a, Mod b) { a.n += b.n; if (a.n >= MOD) a.n -= MOD; return a; }
Mod operator-=(Mod &a, Mod b) { a.n -= b.n; if (a.n < 0) a.n += MOD; return a; }
Mod operator*=(Mod &a, Mod b) { a.n = ((long long)a.n * b.n) % MOD; return a; }
Mod operator+(Mod a, Mod b) { return a += b; }
Mod operator-(Mod a, Mod b) { return a -= b; }
Mod operator*(Mod a, Mod b) { return a *= b; }
Mod operator^(Mod a, int n) {
  if (n == 0) return Mod(1);
  Mod res = (a * a) ^ (n / 2);
  if (n % 2) res = res * a;
  return res;
}

ll inv(ll a, ll p) {
  return (a == 1 ? 1 : (1 - p * inv(p%a, a)) / a + p);
}
Mod operator/(Mod a, Mod b) { return a * Mod(inv(b, MOD)); }

using P = Mod;

vector<P> NTT(P omega, const vector<P> &a) {
  const int n = a.size();
  vector<P> ret = a;
  for (int m = n; m >= 2; m >>= 1, omega *= omega) {
    P rho = 1;
    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] = rho * x;
      }
      rho *= omega;
    }
  }
  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]);
    if (i < b) temp2[i] = P(rhs[i]);
  }
  P root = 3;
  int index = (MOD - 1) / n;
  P omega = root ^ index;
  temp1 = NTT(omega, temp1);
  temp2 = NTT(omega, temp2);
  REP(i,n) temp1[i] *= temp2[i];
  temp1 = NTT(P(1)/omega, temp1);
  vector<ll> ret(n);
  REP(i,n) ret[i] = temp1[i] / P(n);
  return ret;
}

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;
  }
  ll n;
  cin>>n;
  string s;
  cin>>s;
  bitset<300000> bit_i, bit_n;
  REP(i,n) {
    if (s[i] == 'i') bit_i.set(i);
    else bit_n.set(i);
  }
  vecll poly(n+1);
  REP(k,n-1) {
    auto intersect = bit_i & (bit_n >> (k+1));
    poly[k] = intersect.count();
  }
  vecll a(n+1);
  REP(i,n-1) {
    a[i] = poly[i] * fact[i] % MOD;
  }
  vecll b(n+1);
  REP(i,n+1) {
    b[i] = finv[n-i];
  }
  vecll poly2 = convolution(a, b);
  ll ans = 0;
  REP(i,n-1) {
    ll tmp = poly2[i+n] * finv[i] % MOD;
    ans ^= tmp;
  }
  cout<<ans<<"\n";
  return 0;
}
0