結果
問題 | No.287 場合の数 |
ユーザー |
|
提出日時 | 2025-02-03 21:05:31 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 3 ms / 5,000 ms |
コード長 | 1,752 bytes |
コンパイル時間 | 4,487 ms |
コンパイル使用メモリ | 283,952 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2025-02-03 21:05:39 |
合計ジャッジ時間 | 6,077 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
#include <bits/stdc++.h> void fft(std::vector<std::complex<double>>& a, bool invert) { const double PI = acos(-1.0); int n = a.size(); for (int i = 1, j = 0; i < n; ++i) { int bit = n >> 1; for (; j & bit; bit >>= 1) { j -= bit; } j += bit; if (i < j) std::swap(a[i], a[j]); } for (int len = 2; len <= n; len <<= 1) { double angle = 2 * PI / len * (invert ? -1 : 1); std::complex<double> wlen(cos(angle), sin(angle)); for (int i = 0; i < n; i += len) { std::complex<double> w(1); for (int j = 0; j < len / 2; ++j) { std::complex<double> u = a[i + j]; std::complex<double> v = a[i + j + len / 2] * w; a[i + j] = u + v; a[i + j + len / 2] = u - v; w *= wlen; } } } if (invert) { for (std::complex<double> &x : a) { x /= n; } } } std::vector<long long> convolution(const std::vector<long long>& a, const std::vector<long long>& b) { std::vector<std::complex<double>> fa(a.begin(), a.end()), fb(b.begin(), b.end()); int n = 1; while (n < (int)a.size() + (int)b.size() - 1) n <<= 1; fa.resize(n); fb.resize(n); fft(fa, false); fft(fb, false); for (int i = 0; i < n; ++i) { fa[i] *= fb[i]; } fft(fa, true); std::vector<long long> res(n); for (int i = 0; i < n; ++i) { res[i] = llround(fa[i].real()); } return res; } int main() { int n; std::cin >> n; std::vector<long long> a(n + 1, 1); for (int i = 0; i < 3; ++i) { a = convolution(a, a); } std::cout << a[n * 6] << std::endl; }