#766342 (C) No.1958 Bit Game

• 問題: No.1958 Bit Game
• ユーザー: 👑 ygussany
• 提出日時: 2022-05-27 21:44:11
• 言語: C (gcc 12.3.0)
• 結果: AC
• 実行時間: 1,608 ms / 2,000 ms
• コード長: 1,081 bytes
• コンパイル時間: 215 ms
• コンパイル使用メモリ: 33,108 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-09-20 15:40:01
• 合計ジャッジ時間: 33,665 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1,608 ms; 5,248 KB
testcase_01	AC	1,537 ms; 5,376 KB
testcase_02	AC	1,523 ms; 5,376 KB
testcase_03	AC	1,493 ms; 5,376 KB
testcase_04	AC	1,524 ms; 5,376 KB
testcase_05	AC	1,553 ms; 5,376 KB
testcase_06	AC	1,515 ms; 5,376 KB
testcase_07	AC	1,547 ms; 5,376 KB
testcase_08	AC	1,535 ms; 5,376 KB
testcase_09	AC	1,528 ms; 5,376 KB
testcase_10	AC	468 ms; 5,376 KB
testcase_11	AC	748 ms; 5,376 KB
testcase_12	AC	601 ms; 5,376 KB
testcase_13	AC	1,053 ms; 5,376 KB
testcase_14	AC	1,006 ms; 5,376 KB
testcase_15	AC	203 ms; 5,376 KB
testcase_16	AC	90 ms; 5,376 KB
testcase_17	AC	1,259 ms; 5,376 KB
testcase_18	AC	950 ms; 5,376 KB
testcase_19	AC	247 ms; 5,376 KB
testcase_20	AC	1,484 ms; 5,376 KB
testcase_21	AC	389 ms; 5,376 KB
testcase_22	AC	850 ms; 5,376 KB
testcase_23	AC	232 ms; 5,376 KB
testcase_24	AC	1,418 ms; 5,376 KB
testcase_25	AC	420 ms; 5,376 KB
testcase_26	AC	1,310 ms; 5,376 KB
testcase_27	AC	131 ms; 5,376 KB
testcase_28	AC	1,486 ms; 5,376 KB
testcase_29	AC	754 ms; 5,376 KB
testcase_30	AC	1 ms; 5,376 KB
testcase_31	AC	1 ms; 5,376 KB
testcase_32	AC	2 ms; 5,376 KB

ソースコード

#include <stdio.h>

const int Mod = 998244353, bit[19] = {1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072, 262144};

long long pow_mod(int n, long long k)
{
	long long N, ans = 1;
	for (N = n; k > 0; k >>= 1, N = N * N % Mod) if (k & 1) ans = ans * N % Mod;
	return ans;
}

int main()
{
	int i, N, X, Y, A[262144], B[262144];
	scanf("%d %d %d", &N, &X, &Y);
	for (i = 1; i <= X; i++) scanf("%d", &(A[i]));
	for (i = 1; i <= Y; i++) scanf("%d", &(B[i]));
	
	int k, x, y;
	long long ans = 0, tmp;
	for (k = 0; k < 18; k++) {
		for (i = 1, x = 0; i <= X; i++) if ((A[i] & bit[k]) != 0) x++;
		for (i = 1, y = 0; i <= Y; i++) if ((B[i] & bit[k]) != 0) y++;
		if (x == 0 || y == 0) continue;
		for (i = 1, tmp = 0; i < N; i++) tmp += pow_mod(y, i) * pow_mod(Y, N - i - 1) % Mod * (pow_mod(X, N) - pow_mod(X, N - i) * pow_mod(X - x, i) % Mod + Mod) % Mod;
		tmp = (tmp % Mod * (Y - y) + pow_mod(y, N) * (pow_mod(X, N) - pow_mod(X - x, N) + Mod)) % Mod;
		ans += bit[k] * tmp % Mod;
	}
	printf("%lld\n", ans % Mod);
	fflush(stdout);
	return 0;
}