結果

問題 No.1293 2種類の道路
ユーザー 👑 obakyanobakyan
提出日時 2021-07-22 09:05:10
言語 Lua
(LuaJit 2.1.1696795921)
結果
AC  
実行時間 158 ms / 2,000 ms
コード長 1,214 bytes
コンパイル時間 156 ms
コンパイル使用メモリ 6,940 KB
実行使用メモリ 20,228 KB
最終ジャッジ日時 2024-07-17 14:35:30
合計ジャッジ時間 3,963 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 1 ms
6,944 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 1 ms
6,940 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 3 ms
6,944 KB
testcase_09 AC 158 ms
16,092 KB
testcase_10 AC 154 ms
16,260 KB
testcase_11 AC 148 ms
16,072 KB
testcase_12 AC 148 ms
16,112 KB
testcase_13 AC 153 ms
16,164 KB
testcase_14 AC 88 ms
16,788 KB
testcase_15 AC 89 ms
16,920 KB
testcase_16 AC 124 ms
17,536 KB
testcase_17 AC 121 ms
17,788 KB
testcase_18 AC 80 ms
18,788 KB
testcase_19 AC 94 ms
20,220 KB
testcase_20 AC 96 ms
20,228 KB
testcase_21 AC 81 ms
6,940 KB
testcase_22 AC 82 ms
6,940 KB
testcase_23 AC 78 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

local n, d, w = io.read("*n", "*n", "*n")

local function uf_initialize(n)
  local parent = {}
  local rank = {}
  for i = 1, n do
    parent[i] = i
    rank[i] = 1
  end
  return parent, rank
end

local function uf_findroot(idx, parent)
  local idx_update = idx
  while parent[idx] ~= idx do
    idx = parent[idx]
  end
  while parent[idx_update] ~= idx do
    parent[idx_update], idx_update = idx, parent[idx_update]
  end
  return idx
end

local p1 = uf_initialize(n)
local p2, p2rank = uf_initialize(n)

for i = 1, d do
  local a, b = io.read("*n", "*n")
  local ra, rb = uf_findroot(a, p1), uf_findroot(b, p1)
  p1[rb], p1[b] = ra, ra
end
for i = 1, w do
  local a, b = io.read("*n", "*n")
  local ra, rb = uf_findroot(a, p2), uf_findroot(b, p2)
  if ra ~= rb then
    p2rank[ra] = p2rank[ra] + p2rank[rb]
    p2rank[rb] = false
    p2[rb], p2[b] = ra, ra
  end
end

local hub = {}
local ret = {}
for i = 1, n do
  hub[i] = {}
  ret[i] = 0
end

for i = 1, n do
  local a = uf_findroot(i, p1)
  local b = uf_findroot(i, p2)
  if not hub[a][b] then
    hub[a][b] = true
    ret[a] = ret[a] + p2rank[b]
  end
end

local ans = -n
for i = 1, n do
  local a = uf_findroot(i, p1)
  ans = ans + ret[a]
end
print(ans)
0