結果
問題 | No.2670 Sum of Products of Interval Lengths |
ユーザー | suisen |
提出日時 | 2023-10-30 00:48:41 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 61 ms / 2,000 ms |
コード長 | 1,416 bytes |
コンパイル時間 | 2,179 ms |
コンパイル使用メモリ | 117,892 KB |
実行使用メモリ | 9,072 KB |
最終ジャッジ日時 | 2024-02-11 22:15:55 |
合計ジャッジ時間 | 3,942 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,548 KB |
testcase_01 | AC | 61 ms
9,072 KB |
testcase_02 | AC | 30 ms
6,548 KB |
testcase_03 | AC | 9 ms
6,548 KB |
testcase_04 | AC | 29 ms
6,548 KB |
testcase_05 | AC | 58 ms
8,828 KB |
testcase_06 | AC | 59 ms
8,928 KB |
testcase_07 | AC | 30 ms
6,548 KB |
testcase_08 | AC | 8 ms
6,548 KB |
testcase_09 | AC | 30 ms
6,548 KB |
testcase_10 | AC | 59 ms
8,828 KB |
testcase_11 | AC | 59 ms
8,928 KB |
testcase_12 | AC | 58 ms
9,072 KB |
testcase_13 | AC | 59 ms
9,072 KB |
testcase_14 | AC | 59 ms
9,072 KB |
testcase_15 | AC | 59 ms
9,072 KB |
testcase_16 | AC | 61 ms
9,072 KB |
ソースコード
#include <algorithm> #include <iostream> #include <vector> #include <atcoder/modint> #include <atcoder/convolution> using mint = atcoder::modint998244353; using fps = std::vector<mint>; fps fps_inv(const fps& f) { int n = f.size(); fps f_fft, g_fft; fps g{ f[0].inv() }; for (int k = 1; k < n; k *= 2) { f_fft = fps(f.begin(), f.begin() + std::min(2 * k, n)); g_fft = g; f_fft.resize(2 * k), g_fft.resize(2 * k); atcoder::internal::butterfly(f_fft); atcoder::internal::butterfly(g_fft); fps fg(2 * k); for (int i = 0; i < 2 * k; ++i) fg[i] = f_fft[i] * g_fft[i]; atcoder::internal::butterfly_inv(fg); fg.erase(fg.begin(), fg.begin() + k), fg.resize(2 * k); atcoder::internal::butterfly(fg); for (int i = 0; i < 2 * k; ++i) fg[i] *= g_fft[i]; atcoder::internal::butterfly_inv(fg); const mint iz = mint(2 * k).inv(), c = -iz * iz; g.resize(2 * k); for (int i = 0; i < k; ++i) g[k + i] = fg[i] * c; } g.resize(n); return g; } int main() { int n; long long m; std::cin >> n >> m; fps f(n + 1); f[1] = 1; for (int i = 2; i <= n; ++i) f[i] += f[i - 1] - f[i - 2]; for (int i = 1; i <= n; ++i) f[i] *= std::max(0LL, m - i + 1); f[0] = 1; for (int i = 1; i <= n; ++i) f[i] = -f[i]; std::cout << fps_inv(f)[n].val() << std::endl; }