結果

問題 No.1470 Mex Sum
ユーザー daut-dlangdaut-dlang
提出日時 2021-04-09 22:31:25
言語 D
(dmd 2.106.1)
結果
AC  
実行時間 24 ms / 2,000 ms
コード長 2,140 bytes
コンパイル時間 768 ms
コンパイル使用メモリ 120,836 KB
実行使用メモリ 9,380 KB
最終ジャッジ日時 2024-06-22 11:11:54
合計ジャッジ時間 2,812 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 4 ms
6,944 KB
testcase_03 AC 4 ms
6,940 KB
testcase_04 AC 3 ms
6,944 KB
testcase_05 AC 3 ms
6,944 KB
testcase_06 AC 3 ms
6,944 KB
testcase_07 AC 3 ms
6,940 KB
testcase_08 AC 3 ms
6,940 KB
testcase_09 AC 3 ms
6,944 KB
testcase_10 AC 3 ms
6,944 KB
testcase_11 AC 3 ms
6,940 KB
testcase_12 AC 23 ms
8,528 KB
testcase_13 AC 23 ms
7,548 KB
testcase_14 AC 23 ms
9,264 KB
testcase_15 AC 24 ms
7,672 KB
testcase_16 AC 22 ms
7,488 KB
testcase_17 AC 22 ms
7,532 KB
testcase_18 AC 21 ms
9,320 KB
testcase_19 AC 22 ms
8,324 KB
testcase_20 AC 22 ms
7,556 KB
testcase_21 AC 22 ms
8,548 KB
testcase_22 AC 23 ms
9,156 KB
testcase_23 AC 23 ms
7,616 KB
testcase_24 AC 23 ms
9,380 KB
testcase_25 AC 23 ms
8,848 KB
testcase_26 AC 23 ms
8,448 KB
testcase_27 AC 22 ms
8,692 KB
testcase_28 AC 23 ms
9,288 KB
testcase_29 AC 23 ms
8,816 KB
testcase_30 AC 23 ms
7,576 KB
testcase_31 AC 23 ms
9,104 KB
testcase_32 AC 24 ms
7,584 KB
testcase_33 AC 23 ms
9,344 KB
testcase_34 AC 23 ms
7,720 KB
testcase_35 AC 23 ms
7,808 KB
testcase_36 AC 24 ms
8,240 KB
testcase_37 AC 17 ms
7,332 KB
testcase_38 AC 16 ms
7,320 KB
testcase_39 AC 16 ms
7,328 KB
testcase_40 AC 17 ms
8,304 KB
testcase_41 AC 18 ms
9,244 KB
testcase_42 AC 17 ms
8,128 KB
testcase_43 AC 17 ms
9,104 KB
testcase_44 AC 17 ms
7,508 KB
testcase_45 AC 17 ms
7,484 KB
testcase_46 AC 17 ms
7,260 KB
testcase_47 AC 17 ms
7,880 KB
testcase_48 AC 16 ms
8,100 KB
testcase_49 AC 17 ms
7,472 KB
testcase_50 AC 17 ms
7,256 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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)); }

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 A = RDA;

	long cnt1, cnt2;
	foreach (i; 0..N)
	{
		if (A[i] == 1)
			++cnt1;
		else if (A[i] == 2)
			++cnt2;
	}

	auto ans = N * (N-1) / 2;
	ans += cnt1 * cnt2 * 2;
	ans += cnt1 * (N-cnt2-1) - (cnt1*(cnt1-1)/2);

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