#592404 (C++17) No.1504 ヌメロニム

提出ソース

結果

問題	No.1504 ヌメロニム
ユーザー	そすうぽよ
提出日時	2020-12-12 07:27:26
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	690 ms / 2,000 ms
コード長	3,150 bytes
コンパイル時間	1,054 ms
コンパイル使用メモリ	96,204 KB
実行使用メモリ	50,976 KB
最終ジャッジ日時	2023-10-20 10:36:18
合計ジャッジ時間	16,204 ms
ジャッジサーバーID （参考情報）	judge11 / judge14

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	13 ms 12,384 KB
testcase_01	AC	13 ms 12,384 KB
testcase_02	AC	13 ms 12,384 KB
testcase_03	AC	13 ms 12,384 KB
testcase_04	AC	13 ms 12,384 KB
testcase_05	AC	13 ms 12,384 KB
testcase_06	AC	14 ms 12,384 KB
testcase_07	AC	12 ms 12,384 KB
testcase_08	AC	13 ms 12,384 KB
testcase_09	AC	13 ms 12,384 KB
testcase_10	AC	13 ms 12,384 KB
testcase_11	AC	12 ms 12,384 KB
testcase_12	AC	13 ms 12,384 KB
testcase_13	AC	13 ms 12,384 KB
testcase_14	AC	13 ms 12,384 KB
testcase_15	AC	13 ms 12,384 KB
testcase_16	AC	12 ms 12,384 KB
testcase_17	AC	13 ms 12,384 KB
testcase_18	AC	13 ms 12,384 KB
testcase_19	AC	13 ms 12,384 KB
testcase_20	AC	16 ms 13,076 KB
testcase_21	AC	14 ms 12,928 KB
testcase_22	AC	15 ms 13,076 KB
testcase_23	AC	13 ms 12,384 KB
testcase_24	AC	660 ms 50,016 KB
testcase_25	AC	319 ms 33,588 KB
testcase_26	AC	658 ms 50,976 KB
testcase_27	AC	321 ms 35,096 KB
testcase_28	AC	323 ms 35,128 KB
testcase_29	AC	690 ms 50,860 KB
testcase_30	AC	678 ms 50,868 KB
testcase_31	AC	320 ms 34,152 KB
testcase_32	AC	317 ms 35,012 KB
testcase_33	AC	672 ms 50,508 KB
testcase_34	AC	70 ms 18,088 KB
testcase_35	AC	70 ms 17,512 KB
testcase_36	AC	316 ms 31,384 KB
testcase_37	AC	39 ms 15,128 KB
testcase_38	AC	677 ms 50,944 KB
testcase_39	AC	667 ms 50,944 KB
testcase_40	AC	666 ms 50,944 KB
testcase_41	AC	679 ms 50,944 KB
testcase_42	AC	660 ms 50,944 KB
testcase_43	AC	674 ms 50,944 KB
testcase_44	AC	668 ms 50,944 KB
testcase_45	AC	661 ms 50,944 KB
testcase_46	AC	669 ms 50,944 KB
testcase_47	AC	675 ms 50,944 KB
testcase_48	AC	319 ms 34,456 KB
testcase_49	AC	314 ms 33,596 KB
testcase_50	AC	14 ms 13,084 KB
testcase_51	AC	13 ms 12,924 KB
testcase_52	AC	13 ms 12,384 KB
testcase_53	AC	14 ms 12,928 KB
testcase_54	AC	12 ms 12,384 KB
testcase_55	AC	70 ms 17,492 KB
testcase_56	AC	13 ms 12,384 KB
testcase_57	AC	13 ms 12,384 KB
testcase_58	AC	13 ms 12,384 KB
testcase_59	AC	13 ms 12,384 KB
testcase_60	AC	13 ms 12,384 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

// 想定解 ver
#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>;

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;
  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);
  vecll a(n+1);
  REP(i,n-1) {
    a[i] = poly[n-2-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;
}

yukicoder

結果

テストケース

ソースコード