#581863 (D) No.1293 2種類の道路

提出ソース

結果

問題	No.1293 2種類の道路
ユーザー	daut-dlang
提出日時	2020-11-20 22:02:40
言語	D (dmd 2.099.1)
結果	AC
実行時間	152 ms / 2,000 ms
コード長	2,878 bytes
コンパイル時間	1,014 ms
使用メモリ	28,932 KB
最終ジャッジ日時	2023-02-23 18:21:57
合計ジャッジ時間	4,580 ms
ジャッジサーバーID （参考情報）	judge12 / judge13

このコードへのチャレンジ（β）

テストケース

テストケース表示

入力	結果	実行時間使用メモリ
testcase_00	AC	1 ms 6,168 KB
testcase_01	AC	1 ms 6,196 KB
testcase_02	AC	1 ms 6,200 KB
testcase_03	AC	1 ms 6,212 KB
testcase_04	AC	1 ms 6,212 KB
testcase_05	AC	1 ms 6,268 KB
testcase_06	AC	1 ms 6,196 KB
testcase_07	AC	2 ms 6,200 KB
testcase_08	AC	1 ms 6,168 KB
testcase_09	AC	145 ms 12,308 KB
testcase_10	AC	149 ms 12,264 KB
testcase_11	AC	152 ms 12,164 KB
testcase_12	AC	148 ms 12,268 KB
testcase_13	AC	147 ms 12,304 KB
testcase_14	AC	94 ms 15,368 KB
testcase_15	AC	93 ms 15,408 KB
testcase_16	AC	144 ms 16,428 KB
testcase_17	AC	146 ms 18,796 KB
testcase_18	AC	104 ms 21,960 KB
testcase_19	AC	111 ms 28,932 KB
testcase_20	AC	109 ms 28,908 KB
testcase_21	AC	93 ms 6,284 KB
testcase_22	AC	98 ms 6,316 KB
testcase_23	AC	101 ms 6,272 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)); }

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

struct UnionFind(T)
{
	void init(T n) { par = new T[](n); foreach (i; 0..n) par[i] = i; cnt = new T[](n); fill(cnt, 1); }
	T root(T i) { return par[i] == i ? i : (par[i] = root(par[i])); }
	bool same(T i, T j) { return root(i) == root(j); }
	void unite(T i, T j) { i = root(i); j = root(j); if (i == j) return; par[i] = j; cnt[j] += cnt[i]; }
	T size(T i) { return cnt[root(i)]; }
	T[] par, cnt;
}

void main()
{
	auto N = RD;
	auto D = RD;
	auto W = RD;
	UnionFind!long uf1, uf2;
	uf1.init(N);
	uf2.init(N);
	foreach (i; 0..D)
	{
		auto a = RD-1;
		auto b = RD-1;
		uf1.unite(a, b);
	}
	foreach (i; 0..W)
	{
		auto c = RD-1;
		auto d = RD-1;
		uf2.unite(c, d);
	}
	
	auto cnt = new long[](N);
	auto used = new bool[long][](N);
	foreach (i; 0..N)
	{
		auto r1 = uf1.root(i);
		auto r2 = uf2.root(i);
		if (used[r1].get(r2, false) == false)
		{
			used[r1][r2] = true;
			cnt[r1] += uf2.size(r2);
		}
	}

	long ans;
	foreach (i; 0..N)
	{
		auto r1 = uf1.root(i);
		ans += cnt[r1]-1;
	}

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

yukicoder

結果

テストケース

ソースコード