#766613 (D) No.1958 Bit Game

提出ソース

結果

問題	No.1958 Bit Game
ユーザー	daut-dlang
提出日時	2022-05-27 23:10:10
言語	D (dmd 2.107.1)
結果	AC
実行時間	784 ms / 2,000 ms
コード長	3,085 bytes
コンパイル時間	2,587 ms
コンパイル使用メモリ	108,688 KB
実行使用メモリ	20,296 KB
最終ジャッジ日時	2023-09-04 17:18:04
合計ジャッジ時間	20,448 ms
ジャッジサーバーID （参考情報）	judge12 / judge13

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	782 ms 17,760 KB
testcase_01	AC	784 ms 17,684 KB
testcase_02	AC	784 ms 17,740 KB
testcase_03	AC	784 ms 17,792 KB
testcase_04	AC	784 ms 17,688 KB
testcase_05	AC	781 ms 17,964 KB
testcase_06	AC	783 ms 17,780 KB
testcase_07	AC	783 ms 17,712 KB
testcase_08	AC	782 ms 17,764 KB
testcase_09	AC	782 ms 17,848 KB
testcase_10	AC	241 ms 6,408 KB
testcase_11	AC	386 ms 11,264 KB
testcase_12	AC	345 ms 12,148 KB
testcase_13	AC	520 ms 13,028 KB
testcase_14	AC	492 ms 11,368 KB
testcase_15	AC	188 ms 14,776 KB
testcase_16	AC	104 ms 12,492 KB
testcase_17	AC	634 ms 15,992 KB
testcase_18	AC	501 ms 20,296 KB
testcase_19	AC	189 ms 14,392 KB
testcase_20	AC	692 ms 12,548 KB
testcase_21	AC	269 ms 14,824 KB
testcase_22	AC	394 ms 6,600 KB
testcase_23	AC	157 ms 12,868 KB
testcase_24	AC	687 ms 13,344 KB
testcase_25	AC	238 ms 9,504 KB
testcase_26	AC	600 ms 13,840 KB
testcase_27	AC	146 ms 14,844 KB
testcase_28	AC	680 ms 13,176 KB
testcase_29	AC	363 ms 7,092 KB
testcase_30	AC	1 ms 4,380 KB
testcase_31	AC	1 ms 4,380 KB
testcase_32	AC	2 ms 4,380 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

import std.stdio, std.conv, std.functional, std.string;
import std.algorithm, std.array, std.container, std.range, std.typecons;
import std.bigint, std.numeric, std.math, std.random;
import core.bitop;

string FMT_F = "%.10f";

static File _f;
void file_io(string fn) { _f = File(fn, "r"); }
static string[] s_rd;
T _RD(T = long)() { while(!s_rd.length) s_rd = readln.chomp.split; string res = s_rd[0]; s_rd.popFront; return res.to!T; }
T _RD(T = long)(File f) { while(!s_rd.length) s_rd = f.readln.chomp.split; string res = s_rd[0]; s_rd.popFront; return res.to!T; }
T[] _RDA(T = long)(T fix = 0) { auto r = readln.chomp.split.to!(T[]); r[] += fix; return r; }
T[] _RDA(T = long)(File f, T fix = 0) { auto r = f.readln.chomp.split.to!(T[]); r[] += fix; return r; }
T RD(T = long)() { if (_f.isOpen) return _RD!T(_f); else return _RD!T; }
T[] RDA(T = long)(T fix = 0) { if (_f.isOpen)  return _RDA!T(_f, fix); else return _RDA!T(fix); }

size_t[] MAKE_IDX(alias less = "a < b", Range)(Range range) { auto idx = new size_t[](range.length); makeIndex!(less)(range, idx); return idx;}
size_t MIN_POS(alias less = "a < b", Range)(Range range) { auto r = minPos!(less)(range); return range.length - r.length; }

void chmin(T)(ref T x, T y) { x = min(x, y); } void chmax(T)(ref T x, T y) { x = max(x, y); }
bool inside(T)(T x, T b, T e) { return x >= b && x < e; }
T lcm(T)(T x, T y) { return x * (y / gcd(x, y)); }
double euDist(T)(T[] a, T[] b) { auto c = a.dup; c[] -= b[]; c[] *= c[]; return sqrt(cast(double)c.sum); }
double[] rotate(double[] vec, double rad) { return [cos(rad)*vec[0] - sin(rad)*vec[1], sin(rad)*vec[0] + cos(rad)*vec[1]]; }
double norm(double[] vec) { return sqrt(reduce!((a,b)=>a+b*b)(0.0, vec)); }
double dotProd(double[] a, double[] b) { auto r = a.dup; r[] *= b[]; return r.sum; }

//long mod = 10^^9 + 7;
long mod = 998_244_353;
//long mod = 1_000_003;
void moda(ref long x, long y) { x = (x + y) % mod; }
void mods(ref long x, long y) { x = ((x + mod) - (y % mod)) % mod; }
void modm(ref long x, long y) { x = (x * y) % mod; }
void modpow(ref long x, long y) { if (!y) { x = 1; return; } auto t = x; x.modpow(y>>1); x.modm(x); if (y&1) x.modm(t); }
void modd(ref long x, long y) { y.modpow(mod - 2); x.modm(y); }

void main()
{
	auto N = RD;
	auto X = RD;
	auto Y = RD;
	auto A = RDA;
	auto B = RDA;

	auto cnt_a = new long[](18);
	auto cnt_b = new long[](18);
	foreach (i; 0..18)
	{
		auto bit = 1L << i;
		foreach (e; A)
			if (e & bit)
				++cnt_a[i];
		foreach (e; B)
			if (e & bit)
				++cnt_b[i];
	}

	long ans;
	auto aa = new long[](18);
	auto bb = new long[](18);
	foreach (i; 0..18)
	{
		aa[i] = 1;
		bb[i] = cnt_b[i];
	}
	foreach_reverse (j; 0..N)
	{
		long tmp1 = X;
		tmp1.modpow(j);
		long tmp4 = Y;
		tmp4.modpow(j);
		foreach (i; 0..18)
		{
			if (cnt_a[i] == 0 || cnt_b[i] == 0) continue;

			auto tmp = tmp1;
			tmp.modm(aa[i]);
			tmp.modm(cnt_a[i]);

			tmp.modm(bb[i]);
			tmp.modm(tmp4);
			
			tmp.modm(1L<<i);
			ans.moda(tmp);

			aa[i].modm(X - cnt_a[i]);
			bb[i].modm(cnt_b[i]);
		}
	}

	writeln(ans);
	stdout.flush;
	debug readln;
}

yukicoder

結果

テストケース

ソースコード