結果
問題 | No.1757 Many Many Cards |
ユーザー |
![]() |
提出日時 | 2021-11-20 16:42:07 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 135 ms / 2,000 ms |
コード長 | 2,180 bytes |
コンパイル時間 | 3,425 ms |
コンパイル使用メモリ | 215,520 KB |
最終ジャッジ日時 | 2025-01-25 21:54:00 |
ジャッジサーバーID (参考情報) |
judge8 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 36 |
ソースコード
#include<bits/stdc++.h>namespace {#pragma GCC diagnostic push#pragma GCC diagnostic ignored "-Wunused-function"#include<atcoder/all>#pragma GCC diagnostic popusing namespace std;using namespace atcoder;#define rep(i,n)for (int i = 0; i < int(n); ++i)#define rrep(i,n)for (int i = int(n)-1; i >= 0; --i)#define all(x) (x).begin(), (x).end()#define rall(x) (x).rbegin(), (x).rend()template<class T> void chmax(T& a, const T& b) { a = max(a, b); }template<class T> void chmin(T& a, const T& b) { a = min(a, b); }using ll = long long;using P = pair<int,int>;using VI = vector<int>;using VVI = vector<VI>;using VL = vector<ll>;using VVL = vector<VL>;using mint = modint998244353;constexpr int FACT_SIZE = 1000000;mint Fact[FACT_SIZE + 1];mint iFact[FACT_SIZE + 1];const auto fact_init = [] {Fact[0] = mint::raw(1);for(int i = 1; i <= FACT_SIZE; ++i) {Fact[i] = Fact[i-1] * i;}iFact[FACT_SIZE] = Fact[FACT_SIZE].inv();for(int i = FACT_SIZE; i; --i) {iFact[i-1] = iFact[i] * i;}return false;}();mint comb(int n, int k) {if (k == 0) return mint::raw(1);assert(n >= 0 && k >= 0);if (k > n) return mint::raw(0);return Fact[n] * iFact[n - k] * iFact[k];}mint icomb(int n, int k) {return iFact[n] * Fact[n - k] * Fact[k];}mint fact(int n) {return Fact[n];}mint perm(int n, int k) {assert(0 <= n);return Fact[n] * iFact[n - k];}} int main() {ios::sync_with_stdio(false);cin.tie(0);int n, m;cin >> n >> m;// vector<mint> e(m + 1);// rep(i, m + 1) {// if (i % 2 == 0) e[i] = iFact[i];// else e[i] = -iFact[i];// }// vector<mint> f(m + 1);// rep(i, m + 1) {// f[i] = mint(i).pow(2 * n) * iFact[i];// }// f = convolution(f, e);// f.resize(m + 1);mint ans = m * mint(m).pow(2 * n);// rep(i, m + 1) ans += i * f[i] * Fact[i];// remove for tree casefor(int i = 1; i <= m; i++) {if (i - 1 > n) break;mint tmp = i == 1 ? mint(1) : mint(i).pow(i - 2) * Fact[i - 1] * mint(2).pow(i - 1);ans -= comb(m, i) * comb(n, i - 1) * mint(m - i).pow(2 * (n - (i - 1))) * tmp;}cout << ans.val() << '\n';}