結果

問題 No.2513 Power Eraser
ユーザー 👑 amentorimaruamentorimaru
提出日時 2023-06-14 18:16:58
言語 C++17(clang)
(17.0.6 + boost 1.83.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
64,128 KB
testcase_01 AC 2 ms
10,752 KB
testcase_02 AC 2 ms
64,128 KB
testcase_03 AC 7 ms
64,260 KB
testcase_04 AC 47 ms
6,944 KB
testcase_05 AC 10 ms
5,376 KB
testcase_06 AC 73 ms
5,376 KB
testcase_07 AC 32 ms
5,376 KB
testcase_08 AC 58 ms
5,376 KB
testcase_09 AC 91 ms
5,376 KB
testcase_10 AC 57 ms
5,376 KB
testcase_11 AC 9 ms
5,376 KB
testcase_12 AC 43 ms
5,376 KB
testcase_13 AC 1,319 ms
15,104 KB
testcase_14 AC 3,014 ms
27,220 KB
testcase_15 AC 3,466 ms
27,852 KB
testcase_16 AC 1,689 ms
16,128 KB
testcase_17 AC 2,060 ms
17,280 KB
testcase_18 AC 3,076 ms
27,276 KB
testcase_19 AC 4,152 ms
30,288 KB
testcase_20 AC 1,457 ms
15,668 KB
testcase_21 AC 3,181 ms
27,716 KB
testcase_22 AC 3,216 ms
27,584 KB
testcase_23 AC 4,131 ms
30,308 KB
testcase_24 AC 4,191 ms
30,180 KB
testcase_25 AC 4,099 ms
30,304 KB
testcase_26 AC 4,099 ms
30,184 KB
testcase_27 AC 4,153 ms
30,308 KB
testcase_28 AC 4,087 ms
30,180 KB
testcase_29 AC 4,072 ms
30,308 KB
testcase_30 AC 4,123 ms
30,312 KB
testcase_31 AC 4,087 ms
30,308 KB
testcase_32 AC 4,127 ms
30,180 KB
testcase_33 AC 2 ms
5,376 KB
testcase_34 AC 1 ms
5,376 KB
testcase_35 AC 4,133 ms
30,180 KB
testcase_36 AC 4,168 ms
30,308 KB
7_evil_case_1.txt TLE -
7_evil_case_2.txt TLE -
7_evil_case_3.txt TLE -
7_evil_case_4.txt TLE -
7_evil_case_5.txt TLE -
権限があれば一括ダウンロードができます

ソースコード

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