結果

問題 No.2005 Sum of Power Sums
コンテスト
ユーザー KumaTachiRen
提出日時 2023-09-29 10:26:51
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
TLE  
実行時間 -
コード長 2,990 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 4,102 ms
コンパイル使用メモリ 262,504 KB
最終ジャッジ日時 2025-02-17 02:53:44
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12 TLE * 1 -- * 5
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
#include <atcoder/all>

using namespace std;
using namespace atcoder;

struct Fast {
  Fast() {
    std::cin.tie(nullptr);
    ios::sync_with_stdio(false);
    cout << setprecision(10);
  }
} fast;

#define all(a) (a).begin(), (a).end()
#define contains(a, x) ((a).find(x) != (a).end())
#define rep(i, a, b) for (int i = (a); i < (int)(b); i++)
#define rrep(i, a, b) for (int i = (int)(b)-1; i >= (a); i--)
#define writejoin(s, a) rep(_i, 0, (a).size()) cout << (a)[_i] << (_i + 1 < (int)(a).size() ? s : "\n");
#define YN(b) cout << ((b) ? "YES" : "NO") << "\n";
#define Yn(b) cout << ((b) ? "Yes" : "No") << "\n";
#define yn(b) cout << ((b) ? "yes" : "no") << "\n";

using ll = long long;
using vi = vector<int>;
using vvi = vector<vi>;
using vl = vector<ll>;
using mint = modint998244353;
using vm = vector<mint>;

template <typename mint>
class Factorial {
 public:
  Factorial(int max) : n(max) {
    f = vector<mint>(n + 1);
    finv = vector<mint>(n + 1);
    f[0] = 1;
    for (int i = 1; i <= n; i++) f[i] = f[i - 1] * i;
    finv[n] = f[n].inv();
    for (int i = n; i > 0; i--) finv[i - 1] = finv[i] * i;
  }
  mint fact(int k) {
    assert(0 <= k && k <= n);
    return f[k];
  }
  mint fact_inv(int k) {
    assert(0 <= k && k <= n);
    return finv[k];
  }
  mint binom(int k, int r) {
    assert(0 <= k && k <= n);
    if (r < 0 || r > k) return 0;
    return f[k] * finv[r] * finv[k - r];
  }
  mint inv(int k) {
    assert(0 < k && k <= n);
    return finv[k] * f[k - 1];
  }

 private:
  int n;
  vector<mint> f, finv;
};

template <typename mint>
mint bostan_mori(vector<mint> a, vector<mint> b, ll n) {
  if (a.size() < b.size()) a.resize(b.size());
  auto c = vector<mint>(b.size());
  while (n) {
    rep(i, 0, b.size()) c[i] = i % 2 == 0 ? b[i] : -b[i];
    auto a1 = convolution(a, c), b1 = convolution(b, c);
    int r = (int)(n % 2);
    for (int i = 0; i < a.size() && i * 2 + r < a1.size(); i++)
      a[i] = a1[i * 2 + r];
    for (int i = 0; i < b.size(); i++) b[i] = b1[i * 2];
    n /= 2;
  }
  return a[0] / b[0];
};

int main() {
  int n;
  ll m;
  cin >> n >> m;
  const int max = 5000;
  vi c(max + 1, 0);
  rep(i, 0, n) {
    int k;
    cin >> k;
    c[k]++;
  }

  vm p{1}, q{1, -1}, f(max + 1, 0);

  for (int k = 1; k <= max; k++) {
    vm newp(k + 1, 0);
    for (int i = 1; i < k; i++) newp[i] += p[i] * i;
    for (int i = k - 1; i >= 0; i--) newp[i + 1] -= newp[i];
    for (int i = 0; i < k; i++) newp[i + 1] += p[i] * k;
    p = newp;
    vm newq(k + 2, 0);
    for (int i = 0; i < k + 1; i++) {
      newq[i] += q[i];
      newq[i + 1] -= q[i];
    }
    q = newq;
    for (int i = max; i > 0; i--) f[i] -= f[i - 1];
    if (c[k] > 0) {
      rep(i, 0, p.size()) f[i] += c[k] * p[i];
    }
  }
  Factorial<mint> fact(n);
  vm r(n + 1, 0);
  rep(i, 0, n + 1) {
    r[i] = fact.binom(n, i);
    if (i & 1) r[i] = -r[i];
  }
  q = convolution(q, r);
  mint ans = bostan_mori<mint>(f, q, m);
  cout << ans.val() << endl;
}
0