結果

問題 No.1390 Get together
ユーザー 👑 obakyan
提出日時 2021-05-24 08:29:49
言語 Lua
(LuaJit 2.1.1734355927)
結果
AC  
実行時間 200 ms / 2,000 ms
コード長 764 bytes
コンパイル時間 109 ms
コンパイル使用メモリ 6,948 KB
実行使用メモリ 25,216 KB
最終ジャッジ日時 2024-10-12 14:25:35
合計ジャッジ時間 5,794 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

local n, m = io.read("*n", "*n")
local col = {}

for i = 1, n do
  col[i] = {}
end
for i = 1, n do
  local b, c = io.read("*n", "*n")
  table.insert(col[c], b)
end

local parent = {}
for i = 1, m do
  parent[i] = i
end

local function uf_findroot(idx)
  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 ret = 0
for icol = 1, n do
  if 1 < #col[icol] then
    local r = uf_findroot(col[icol][1])
    for i = 2, #col[icol] do
      local b = col[icol][i]
      local rb = uf_findroot(b)
      if rb ~= r then
        ret = ret + 1
        parent[rb], parent[b] = r, r
      end
    end
  end
end
print(ret)
0