結果

問題 No.2763 Macaron Gift Box
ユーザー startcppstartcpp
提出日時 2024-05-17 21:49:54
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
MLE  
実行時間 -
コード長 1,138 bytes
コンパイル時間 1,924 ms
コンパイル使用メモリ 114,860 KB
実行使用メモリ 814,040 KB
最終ジャッジ日時 2024-05-17 21:49:58
合計ジャッジ時間 3,688 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 1 ms
6,940 KB
testcase_04 AC 1 ms
6,940 KB
testcase_05 AC 1 ms
6,940 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 MLE -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

//項の数がO(NlogN)なのですが、分割FFTは...O(N^2)のメモリを使用するのでダメですねorz
#include <iostream>
#include <atcoder/modint>
#include <atcoder/convolution>
#define int long long
#define rep(i, n) for(i = 0; i < n; i++)
using namespace std;
using namespace atcoder;
using mint = modint998244353;

//Π f_i(x) mod x^(n + 1) を求める, i in [l, r), r - l >= 1, f_i(x) mod x^(n + 1) = f_i(x) とする。
vector<mint> multiply(vector<vector<mint>> &f, int l, int r, int n) {
	if (r - l == 1) {
		return f[l];
	}
	vector<mint> res1 = multiply(f, l, (l + r) / 2, n);
	vector<mint> res2 = multiply(f, (l + r) / 2, r, n);
	vector<mint> ret = convolution(res1, res2);
	if (ret.size() > n + 1) ret.resize(n + 1);
	return ret;
}

signed main() {
	int n, K;
	cin >> n >> K;

	vector<vector<mint>> fs;
	for (int i = 1; i <= n; i++) {
		vector<mint> f(n + 1);
		for (int j = 0; i * j <= n && j <= K; j++) {
			f[i * j] = 1;
		}
		fs.push_back(f);
	}

	vector<mint> res = multiply(fs, 0, n, n + 1);
	for (int i = 1; i <= n; i++) {
		cout << res[i].val();
		if (i + 1 <= n) cout << " ";
	}
	cout << endl;
	return 0;
}
0