結果

問題 No.1753 Don't cheat.
ユーザー nok0nok0
提出日時 2021-06-19 01:17:57
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 756 ms / 3,000 ms
コード長 1,113 bytes
コンパイル時間 2,392 ms
コンパイル使用メモリ 209,348 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-10 06:46:13
合計ジャッジ時間 28,351 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 752 ms
6,812 KB
testcase_01 AC 751 ms
6,944 KB
testcase_02 AC 755 ms
6,940 KB
testcase_03 AC 751 ms
6,944 KB
testcase_04 AC 751 ms
6,948 KB
testcase_05 AC 752 ms
6,944 KB
testcase_06 AC 754 ms
6,940 KB
testcase_07 AC 752 ms
6,940 KB
testcase_08 AC 746 ms
6,940 KB
testcase_09 AC 747 ms
6,944 KB
testcase_10 AC 750 ms
6,940 KB
testcase_11 AC 748 ms
6,940 KB
testcase_12 AC 746 ms
6,940 KB
testcase_13 AC 749 ms
6,940 KB
testcase_14 AC 750 ms
6,944 KB
testcase_15 AC 754 ms
6,940 KB
testcase_16 AC 750 ms
6,940 KB
testcase_17 AC 754 ms
6,940 KB
testcase_18 AC 756 ms
6,940 KB
testcase_19 AC 749 ms
6,940 KB
testcase_20 AC 754 ms
6,944 KB
testcase_21 AC 752 ms
6,944 KB
testcase_22 AC 753 ms
6,940 KB
testcase_23 AC 754 ms
6,940 KB
testcase_24 AC 753 ms
6,944 KB
testcase_25 AC 755 ms
6,940 KB
testcase_26 AC 756 ms
6,940 KB
testcase_27 AC 753 ms
6,944 KB
testcase_28 AC 749 ms
6,940 KB
testcase_29 AC 751 ms
6,944 KB
testcase_30 AC 749 ms
6,944 KB
testcase_31 AC 748 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#include <atcoder/modint>

using mint = atcoder::modint998244353;

template <class T>
void fwt(std::vector<T> &f) {
	int n = f.size();
	for(int i = 1; i < n; i <<= 1) {
		for(int j = 0; j < n; j++) {
			if((j & i) == 0) {
				T x = f[j], y = f[j | i];
				f[j] = x + y, f[j | i] = x - y;
			}
		}
	}
}

template <class T>
void ifwt(std::vector<T> &f) {
	int n = f.size();
	const T tinv = T(1) / 2;
	for(int i = 1; i < n; i <<= 1) {
		for(int j = 0; j < n; j++) {
			if((j & i) == 0) {
				T x = f[j], y = f[j | i];
				f[j] = (x + y) * tinv, f[j | i] = (x - y) * tinv;
			}
		}
	}
}

constexpr int sz = 1 << 11;
mint res;
int main() {
	int n;
	cin >> n;
	std::vector<int> a(sz);
	for(int i = 0; i <= n; i++) cin >> a[i];
	int sum = accumulate(a.begin(), a.end(), 0);
	vector<mint> f(sz);
	f[0] = 1;
	for(int i = 1; i < sz; i++) f[i] = -mint(a[i]) / sum;
	for(int x = 1; x < sz; x++) {
		auto g = f;
		g[x] = 0;
		fwt(g);
		for(int i = 0; i < sz; i++) g[i] = g[i].inv();
		ifwt(g);
		res += g[x];
	}
	res *= a[0];
	res /= sum;
	cout << res.val() << endl;
	return 0;
}
0