結果

問題 No.1958 Bit Game
コンテスト
ユーザー startcpp
提出日時 2022-05-28 02:53:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 969 bytes
コンパイル時間 3,155 ms
コンパイル使用メモリ 143,960 KB
実行使用メモリ 26,752 KB
最終ジャッジ日時 2024-09-20 20:09:31
合計ジャッジ時間 17,410 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 11 WA * 17 RE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

//ビットごとにできるので、はい。
#include <iostream>
#include <atcoder/all>
#include <vector>
#define rep(i, n) for(i = 0; i < n; i++)
using namespace std;
using namespace atcoder;
using mint = modint998244353;
typedef vector<mint> Vec;
typedef vector<Vec> Mat;

int n, X, Y;
int a[200000], b[200000];

int main() {
	int i, j;
	cin >> n >> X >> Y;
	rep(i, X) cin >> a[i];
	rep(i, Y) cin >> b[i];

	mint ans = 0;
	rep(i, 18) {
		int oneA = 0, oneB = 0;
		rep(j, X) if ((a[j] >> i) % 2) oneA++;
		rep(j, Y) if ((b[j] >> i) % 2) oneB++;
		
		Mat dp(2 * n + 1, Vec(2, 0));
		dp[0][0] = 1;
		rep(j, 2 * n) {
			if (j % 2 == 0) {
				dp[j + 1][0] += (X - oneA) * dp[j][0];
				dp[j + 1][1] += oneA * dp[j][0];
				dp[j + 1][1] += X * dp[j][1];
			}
			else {
				dp[j + 1][0] += Y * dp[j][0];
				dp[j + 1][0] += (Y - oneB) * dp[j][1];
				dp[j + 1][1] += oneB * dp[j][1];
			}
		}
		ans += (1 << i) * dp[2 * n][1];
	}

	cout << ans.val() << endl;
	return 0;
}
0