結果

問題 No.3538 Not First Place
コンテスト
ユーザー V_Melville
提出日時 2026-05-08 22:55:13
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 1,293 ms / 5,000 ms
コード長 1,482 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 4,300 ms
コンパイル使用メモリ 377,796 KB
実行使用メモリ 139,412 KB
最終ジャッジ日時 2026-05-08 22:55:27
合計ジャッジ時間 10,299 ms
ジャッジサーバーID
(参考情報)
judge3_0 / judge1_1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
#define rep(i, n) for (int i = 0; i < (n); ++i)

using namespace std;
using mint = modint998244353;

struct modinv {
  int n; vector<mint> d;
  modinv(): n(2), d({0,1}) {}
  mint operator()(int i) {
    while (n <= i) d.push_back(-d[mint::mod()%n]*(mint::mod()/n)), ++n;
    return d[i];
  }
  mint operator[](int i) const { return d[i];}
} invs;
struct modfact {
  int n; vector<mint> d;
  modfact(): n(2), d({1,1}) {}
  mint operator()(int i) {
    while (n <= i) d.push_back(d.back()*n), ++n;
    return d[i];
  }
  mint operator[](int i) const { return d[i];}
} facts;
struct modfactinv {
  int n; vector<mint> d;
  modfactinv(): n(2), d({1,1}) {}
  mint operator()(int i) {
    while (n <= i) d.push_back(d.back()*invs(n)), ++n;
    return d[i];
  }
  mint operator[](int i) const { return d[i];}
} ifacts;
mint comb(int n, int k) {
  if (n < k || k < 0) return 0;
  return facts(n)*ifacts(k)*ifacts(n-k);
}

mint f(int c, int k, int u) {
    mint res;
    int r = min(c, k/(u+1));
    rep(j, r+1) {
        mint now = comb(c, j)*comb(k-j*(u+1)+c-1, c-1);
        if (j&1) res -= now;
        else res += now;
    }
    return res;
}

int main() {
    int n, m, k, l;
    cin >> n >> m >> k >> l;
    
    mint ans;
    int tot = m*k;
    
    for (int x = l; x <= m; ++x) {
        int r = tot-x;
        ans += f(n-1, r, m) - f(n-1, r, x);
    }
    
    cout << ans.val() << '\n';
    
    return 0;
}
0