結果

問題 No.2062 Sum of Subset mod 999630629
ユーザー shobonvip
提出日時 2022-08-26 23:43:37
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 893 bytes
コンパイル時間 4,699 ms
コンパイル使用メモリ 262,872 KB
最終ジャッジ日時 2025-01-31 05:51:53
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

typedef modint998244353 mint;
typedef long long ll;

int main(){
	// sum(A[i])はかなり小さいので, デカイやつは一周まわるので除外しまくる.
	int N;
	cin >> N;
	vector<int> A(N);
	
	mint ans = 0;
	mint nnv = mint(2).pow(N-1);
	queue<vector<mint>> Q;
	int asum = 0;

	for (int i=0; i<N; i++){
		cin >> A[i];
		ans += nnv * A[i];
		Q.push({A[i], 1});
		asum += A[i];
	}
	
	// これは多項式をマージするやつのテク O(N (log N)^2)
	while (Q.size() > 1){
		vector<mint> X = Q.front(); Q.pop();
		vector<mint> Y = Q.front(); Q.pop();
		Q.push(convolution<mint>(X, Y));
	}

	vector<mint> W = Q.front();
	mint jogai = 0;

	for (int i=0; i<=N; i++){
		if (asum - i >= 999630629){
			jogai += W[i];
		}
	}

	ans -= jogai * mint(999630629);
	cout << ans.val() << endl;

}
0