結果

問題 No.2381 Gift Exchange Party
ユーザー _____TAB__________TAB_____
提出日時 2023-07-15 13:16:53
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 44 ms / 2,000 ms
コード長 2,946 bytes
コンパイル時間 877 ms
コンパイル使用メモリ 75,192 KB
実行使用メモリ 6,312 KB
最終ジャッジ日時 2023-10-15 04:12:50
合計ジャッジ時間 2,328 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,372 KB
testcase_01 AC 2 ms
4,368 KB
testcase_02 AC 3 ms
4,372 KB
testcase_03 AC 2 ms
4,372 KB
testcase_04 AC 4 ms
4,664 KB
testcase_05 AC 7 ms
6,076 KB
testcase_06 AC 4 ms
4,404 KB
testcase_07 AC 4 ms
4,436 KB
testcase_08 AC 6 ms
5,984 KB
testcase_09 AC 2 ms
4,368 KB
testcase_10 AC 4 ms
4,372 KB
testcase_11 AC 5 ms
4,816 KB
testcase_12 AC 2 ms
4,368 KB
testcase_13 AC 3 ms
4,372 KB
testcase_14 AC 5 ms
5,456 KB
testcase_15 AC 4 ms
4,372 KB
testcase_16 AC 4 ms
4,368 KB
testcase_17 AC 6 ms
5,404 KB
testcase_18 AC 4 ms
4,440 KB
testcase_19 AC 4 ms
4,612 KB
testcase_20 AC 43 ms
6,136 KB
testcase_21 AC 7 ms
6,312 KB
testcase_22 AC 44 ms
6,156 KB
testcase_23 AC 44 ms
6,144 KB
testcase_24 AC 4 ms
4,372 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
using namespace std;

#include <iostream>
#include <cassert>

template<long long mod>
class modint{
private:
  using T = long long;
  T a;
public:
  constexpr modint(const long long x = 0) noexcept : a((x%mod+mod)%mod) {}
  constexpr T& value() noexcept { return a; }
  constexpr const T& value() const noexcept { return a; }
  constexpr modint operator-() const noexcept {
    return modint(0) -= *this;
  }
  constexpr modint operator+(const modint& rhs) const noexcept {
    return modint(*this) += rhs;
  }
  constexpr modint operator-(const modint& rhs) const noexcept {
    return modint(*this) -= rhs;
  }
  constexpr modint operator*(const modint& rhs) const noexcept {
    return modint(*this) *= rhs;
  }
  constexpr modint operator/(const modint& rhs) const noexcept {
    return modint(*this) /= rhs;
  }
  constexpr modint& operator+=(const modint& rhs) noexcept {
    a += rhs.a;
    if(a >= mod) a -= mod;
    return *this;
  }
  constexpr modint& operator-=(const modint& rhs) noexcept {
    if(a < rhs.a) a += mod;
    a -= rhs.a;
    return *this;
  }
  constexpr modint& operator*=(const modint& rhs) noexcept {
    a = a*rhs.a%mod;
    return *this;
  }
  constexpr modint& operator/=(const modint& rhs) noexcept {
    return *this *= rhs.inv();
  }
  constexpr bool operator==(const modint& rhs) const noexcept {
    return a == rhs.a;
  }
  constexpr bool operator!=(const modint& rhs) const noexcept {
    return not (*this == rhs);
  }
  constexpr modint pow(long long k) const noexcept {
    modint ret(1);
    modint x = k > 0 ? *this : this->inv();
    k = abs(k);
    while(k > 0){
      if(k&1) ret *= x;
      x *= x;
      k >>= 1;
    }
    return ret;
  }
  constexpr modint inv() const noexcept {
    return pow(mod-2);
  }
  friend std::ostream& operator<<(std::ostream &os, const modint &X) noexcept {
    return os << X.a;
  }
  friend std::istream& operator>>(std::istream &is, modint &X) noexcept {
    is >> X.a;
    X.a %= mod;
    if(X.a < 0) X.a += mod;
    return is;
  }
};

#include <vector>
#include <cassert>

template<typename T>
struct Combination {
private:
  int sz;
  vector<T> F, F_;
public:
  Combination(int sz) : sz(sz), F(sz+1), F_(sz+1) {
    F[0] = 1;
    for(int i = 0; i < sz; ++i) F[i+1] = F[i]*(i+1);
    F_.back() = (T)1/F.back();
    for(int i = sz-1; i >= 0; --i) F_[i] = F_[i+1]*(i+1);
  }
  T C(int n, int k){
    assert(n <= sz);
    if(n < 0 or k > n) return (T)0;
    return F[n]*F_[k]*F_[n-k];
  }
  T P(int n, int k){
    assert(n <= sz);
    if(n < 0 or k > n) return (T)0;
    return F[n]*F_[n-k];
  }
  T factorial(int k){
    return F[k];
  }
};

int main(){
  using mint = modint<998244353>;
  int n, p;
  cin >> n >> p;
  mint ans = 1;
  for(int i = 1; i <= n; ++i)
    ans *= i;
  Combination<mint> comb(n);
  for(int i = 0; i <= n/p; ++i){
    mint t = comb.P(n,i*p)/comb.factorial(i)/mint(p).pow(i);
    ans -= t;
  }
  cout << ans << '\n';
}
0