結果

問題 No.2513 Power Eraser
コンテスト
ユーザー amentorimaru
提出日時 2023-06-14 18:16:58
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
AC  
実行時間 4,191 ms / 6,000 ms
コード長 2,607 bytes
コンパイル時間 2,855 ms
コンパイル使用メモリ 143,236 KB
実行使用メモリ 64,260 KB
最終ジャッジ日時 2024-09-20 16:34:33
合計ジャッジ時間 115,969 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using mint2 = modint998244353;
using ll = long long;
#define FOR(i, a, b) for(ll i=(a); i<(b);i++)
#define REP(i, n) for(ll i=0; i<(n);i++)
#define ROF(i, a, b) for(ll i=(b-1); i>=(a);i--)
#define PER(i, n) for(ll i=n-1; i>=0;i--)
#define VL vector<ll>
#define all(i) begin(i),end(i)
template<typename T = ll>
vector<T> read(size_t n) {
  vector<T> ts(n);
  for (size_t i = 0; i < n; i++) cin >> ts[i];
  return ts;
}

ll n;
VL a;

vector<mint2> rev(vector<mint2>& f, ll mod) {
  vector<mint2> res = { mint2(1) / f[0] };
  ll t = 1;
  while (t < mod) {
    t *= 2;
    auto m = convolution(res, res);
    m = convolution(m, f);
    vector<mint2> nres(t);
    REP(i, t) {
      if (i < res.size())
        nres[i] += 2 * res[i];
      if (i < m.size())
        nres[i] += -m[i];
    }
    res = nres;
  }
  return res;
}


vector<mint2> div(vector<mint2>& f, vector<mint2>& g) {
  auto rf = f;
  auto rg = g;
  reverse(all(rf));
  reverse(all(rg));
  auto q = convolution(rf, rev(rg, f.size() - g.size() + 1));
  while (q.size() > f.size() - g.size() + 1) {
    q.pop_back();
  }
  reverse(all(q));
  auto gq = convolution(g, q);
  auto res = f;
  REP(i, res.size()) {
    if (i < gq.size()) {
      res[i] -= gq[i];
    }
  }
  while (res.back() == 0 && res.size() > 1) {    
    res.pop_back();
  }
  return res;
}

mint2 solve3(ll l1, ll l2, ll d) {
  ll c = 1;
  while (c < d) {
    c *= 2;
  }  
  VL a1(d), a2(d);
  REP(i, d) {
    a2[i] = a[l1 + i];
    a1[i] = a[l2 + i];
  }
  vector<vector<mint2>> f1(c * 2);
  vector<vector<mint2>> f2(c * 2);
  FOR(i, c, c * 2) {
    if (i < c + d) {
      f1[i].assign(2, 1);
      f1[i][0] = -a1[i - c];
      f2[i].assign(2, 1);
      f2[i][0] = -a2[i - c];
    }
    else {
      f1[i].assign(1, 1);
      f2[i].assign(1, 1);
    }
  }
  PER(i, c) {
    f1[i] = convolution(f1[i * 2], f1[i * 2 + 1]);
    f2[i] = convolution(f2[i * 2], f2[i * 2 + 1]);
  }
  f1[1] = div(f1[1], f2[1]);
  FOR(i, 2, 2 * c) {
    auto& p = f1[i / 2];
    if (f2[i].size() != 1)
      f1[i] = div(p, f2[i]);
  }

  mint2 res = 1;
  FOR(i, c, 2 * c) {
    res *= f1[i][0];
  }
  return res;
}

mint2 solve2(ll l, ll r) {
  if (l >= r - 1)
    return 1;
  mint2 res = 1;
  ll d = (r - l) / 2;
  res *= solve3(l, r - d, d);
  res *= solve2(l, r - d);
  res *= solve2(l + d, r);
  return res;
}

void solve() {
  cin >> n;
  a = read(n);
  mint2 ans = 1;
  cout << solve2(0, n).val();
}


int main() {
  ll t = 1;
  //cin >> t;
  while (t--) {
    solve();
  }
  return 0;
}
0