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

提出ソース

結果

問題	No.1504 ヌメロニム
ユーザー	そすうぽよ
提出日時	2020-12-12 07:27:26
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	663 ms / 2,000 ms
コード長	3,150 bytes
コンパイル時間	1,062 ms
コンパイル使用メモリ	96,556 KB
実行使用メモリ	51,052 KB
最終ジャッジ日時	2024-09-20 06:10:08
合計ジャッジ時間	14,811 ms
ジャッジサーバーID （参考情報）	judge1 / judge3

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	11 ms 12,672 KB
testcase_01	AC	10 ms 12,412 KB
testcase_02	AC	11 ms 12,628 KB
testcase_03	AC	11 ms 12,544 KB
testcase_04	AC	11 ms 12,584 KB
testcase_05	AC	11 ms 12,544 KB
testcase_06	AC	11 ms 12,532 KB
testcase_07	AC	10 ms 12,412 KB
testcase_08	AC	11 ms 12,476 KB
testcase_09	AC	10 ms 12,544 KB
testcase_10	AC	10 ms 12,672 KB
testcase_11	AC	11 ms 12,544 KB
testcase_12	AC	10 ms 12,672 KB
testcase_13	AC	11 ms 12,544 KB
testcase_14	AC	11 ms 12,672 KB
testcase_15	AC	11 ms 12,560 KB
testcase_16	AC	10 ms 12,544 KB
testcase_17	AC	12 ms 12,476 KB
testcase_18	AC	11 ms 12,544 KB
testcase_19	AC	11 ms 12,544 KB
testcase_20	AC	12 ms 12,908 KB
testcase_21	AC	11 ms 12,896 KB
testcase_22	AC	13 ms 12,796 KB
testcase_23	AC	11 ms 12,616 KB
testcase_24	AC	625 ms 50,172 KB
testcase_25	AC	311 ms 33,632 KB
testcase_26	AC	619 ms 50,832 KB
testcase_27	AC	298 ms 34,972 KB
testcase_28	AC	295 ms 35,028 KB
testcase_29	AC	619 ms 50,644 KB
testcase_30	AC	643 ms 50,656 KB
testcase_31	AC	307 ms 34,192 KB
testcase_32	AC	296 ms 34,852 KB
testcase_33	AC	663 ms 50,428 KB
testcase_34	AC	66 ms 18,004 KB
testcase_35	AC	62 ms 17,436 KB
testcase_36	AC	295 ms 31,324 KB
testcase_37	AC	36 ms 15,096 KB
testcase_38	AC	656 ms 50,824 KB
testcase_39	AC	651 ms 50,916 KB
testcase_40	AC	634 ms 50,988 KB
testcase_41	AC	629 ms 50,916 KB
testcase_42	AC	639 ms 50,968 KB
testcase_43	AC	646 ms 50,764 KB
testcase_44	AC	641 ms 50,916 KB
testcase_45	AC	629 ms 50,920 KB
testcase_46	AC	621 ms 50,916 KB
testcase_47	AC	617 ms 51,052 KB
testcase_48	AC	303 ms 34,248 KB
testcase_49	AC	296 ms 33,660 KB
testcase_50	AC	14 ms 12,880 KB
testcase_51	AC	11 ms 12,800 KB
testcase_52	AC	10 ms 12,544 KB
testcase_53	AC	11 ms 12,804 KB
testcase_54	AC	11 ms 12,544 KB
testcase_55	AC	64 ms 17,348 KB
testcase_56	AC	12 ms 12,416 KB
testcase_57	AC	11 ms 12,476 KB
testcase_58	AC	11 ms 12,544 KB
testcase_59	AC	11 ms 12,672 KB
testcase_60	AC	10 ms 12,544 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

結果

テストケース

ソースコード