結果

問題 No.1875 Flip Cards
ユーザー shino16shino16
提出日時 2022-03-05 06:41:35
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 4,199 bytes
コンパイル時間 3,106 ms
コンパイル使用メモリ 226,032 KB
最終ジャッジ日時 2024-11-15 02:10:29
合計ジャッジ時間 3,601 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
main.cpp: In function 'std::vector<atcoder::static_modint<998244353> > exp(std::vector<atcoder::static_modint<998244353> >, int)':
main.cpp:58:35: error: 'ceil_pow2' is not a member of 'atcoder::internal'
   58 | vector<mint> inv(vector<mint> f, int deg = poly_deg) {
      |                                   ^~~~~~~~~

ソースコード

diff # 

// 制約チェック

#line 1 "main.cpp"
#include <atcoder/convolution>
#include <atcoder/modint>

#line 2 "lib/prelude.hpp"
#ifndef LOCAL
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2")
#endif
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define rep2(i, m, n) for (auto i = (m); i < (n); i++)
#define rep(i, n) rep2(i, 0, n)
#define repr2(i, m, n) for (auto i = (n); i-- > (m);)
#define repr(i, n) repr2(i, 0, n)
#define all(x) begin(x), end(x)
#line 5 "main.cpp"
using mint = atcoder::modint998244353;

int poly_deg = -1;

vector<mint> add(vector<mint> a, vector<mint> b) {
  int n = max(a.size(), b.size());
  vector<mint> res(n);
  rep(i, n) res[i] = (i < a.size() ? a[i] : 0) + (i < b.size() ? b[i] : 0);
  return res;
}

vector<mint> mul(mint r, vector<mint> a) {
  for (auto& e : a) e *= r;
  return a;
}

vector<mint> diff(vector<mint> a) {
  rep(i, a.size() - 1) a[i] = a[i + 1] * (i + 1);
  a.pop_back();
  return a;
}

vector<mint> integr(vector<mint> a) {
  // a.push_back(0);
  repr(i, a.size() - 1) a[i + 1] = a[i] / (i + 1);
  a[0] = 0;
  return a;
}

vector<mint> conv(vector<mint> a, vector<mint> b, int deg = poly_deg) {
  if (deg != -1)
    a.resize(min<int>(a.size(), deg)), b.resize(min<int>(b.size(), deg));
  auto f = convolution(a, b);
  if (deg != -1) f.resize(deg);
  return f;
}

vector<mint> inv(vector<mint> f, int deg = poly_deg) {
  assert(f[0] != 0);
  vector<mint> g = {f[0].inv()};
  for (int m = 1; m < deg; m *= 2) {
    auto h = conv(f, g, m * 2);
    fill(h.begin(), h.begin() + m, 0);
    auto hg = conv(h, g, m * 2);
    rep(i, m * 2) hg[i] = -hg[i];
    g.insert(g.end(), hg.begin() + m, hg.end());
  }
  g.resize(deg);
  return g;
}

vector<mint> exp(vector<mint> h, int deg = poly_deg) {
  assert(h[0] == 0);
  int n = 1 << atcoder::internal::ceil_pow2(deg);
  h.resize(n);
  vector<mint> f = {1}, g = {1};
  vector<mint> q(h.size() - 1);
  rep(i, q.size()) q[i] = h[i + 1] * (i + 1);
  for (int m = 1; m < n; m *= 2) {
    auto fgg = conv(conv(f, g, m), g, m);
    g.resize(m);
    rep(i, m) g[i] = g[i] * 2 - fgg[i];
    auto fq = conv(f, {q.begin(), q.begin() + m - 1}, m * 2 - 1);
    rep(i, fq.size()) fq[i] =
        -fq[i] + (i < f.size() - 1 ? f[i + 1] * (i + 1) : 0);
    auto w = conv(g, fq, m * 2 - 1);
    rep(i, m - 1) w[i] += q[i];
    w.push_back(0);
    repr(i, m * 2 - 1) w[i + 1] = h[i + 1] - w[i] / (i + 1);
    w[0] = h[0];
    auto fw = conv(f, w, m * 2);
    f.resize(m * 2);
    rep(i, m * 2) f[i] += fw[i];
  }
  f.resize(deg);
  return f;
}

vector<mint> taylor_shift(vector<mint> a, int c) {
  int n = a.size();
  vector<mint> fact(n);
  fact[0] = 1;
  rep(i, n - 1) a[i + 1] *= fact[i + 1] = fact[i] * (i + 1);
  reverse(all(a));
  a = conv(move(a), exp(vector<mint>{0, c}, n), n);
  a.resize(n);
  reverse(all(a));
  rep(i, n) a[i] /= fact[i];
  return a;
}

struct ratio_t {
  vector<mint> numer, denom;
};

ratio_t total(vector<ratio_t> qs) {
  int n = qs.size();
  qs.resize(n * 2);
  move(qs.begin(), qs.begin() + n, qs.begin() + n);
  repr2(i, 1, n) qs[i] =
      ratio_t{add(conv(qs[i * 2].numer, qs[i * 2 + 1].denom, -1),
                  conv(qs[i * 2].denom, qs[i * 2 + 1].numer, -1)),
              conv(qs[i * 2].denom, qs[i * 2 + 1].denom, -1)};
  return qs[1];
}

vector<mint> eval(ratio_t q, int deg = poly_deg) {
  return conv(move(q.numer), inv(move(q.denom), deg), deg);
}

int n, m, a[100000], b[100000], c[100000];

int main() {
  scanf("%d%d", &n, &m);
  assert(1 <= n && n <= 100000);
  assert(1 <= m && m <= 100000);
  rep(i, n)
    scanf("%d%d%d", a + i, b + i, c + i),
    assert(1 <= a[i] && a[i] <= 998244352),
    assert(1 <= b[i] && b[i] <= 998244352),
    assert(1 <= c[i] && c[i] <= 998244352);

  poly_deg = m+1;

  mint ap = 1;
  rep(i, n) ap *= mint(a[i]).pow(c[i]);

  vector<ratio_t> summands(n);
  rep(i, n) summands[i] =
      ratio_t{{mint(1) * c[i] * b[i] / a[i]}, {1, mint(1) * b[i] / a[i]}};

  auto f = mul(ap, exp(integr(eval(total(summands)))));
  f = taylor_shift(move(f), -1);

  summands.resize(m + 1);
  rep(i, m + 1) summands[i] = ratio_t{{f[i]}, {1, -i}};
  f = eval(total(summands));

  rep2(k, 1, m+1) printf("%d\n", f[k].val());
}
0