結果
問題 | No.2316 Freight Train |
ユーザー |
👑 |
提出日時 | 2023-07-16 16:51:16 |
言語 | Lua (LuaJit 2.1.1734355927) |
結果 |
AC
|
実行時間 | 197 ms / 2,000 ms |
コード長 | 912 bytes |
コンパイル時間 | 122 ms |
コンパイル使用メモリ | 5,248 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-09-17 13:01:02 |
合計ジャッジ時間 | 6,478 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
local UF = {}UF.create = function(self, n)self.parent = {}for i = 1, n doself.parent[i] = iendendUF.getroot = function(self, idx)local parent = self.parentlocal idx_update = idxwhile parent[idx] ~= idx doidx = parent[idx]endwhile parent[idx_update] ~= idx doparent[idx_update], idx_update = idx, parent[idx_update]endreturn idxendUF.unite = function(self, a, b)local ra = self:getroot(a)local rb = self:getroot(b)self.parent[b], self.parent[rb] = ra, raendUF.new = function(n)local obj = {}setmetatable(obj, {__index = UF})obj:create(n)return objendlocal n, q = io.read("*n", "*n")local uf = UF.new(n)for i = 1, n dolocal p = io.read("*n")if 0 < p thenuf:unite(p, i)endendfor iq = 1, q dolocal a, b = io.read("*n", "*n")local ra = uf:getroot(a)local rb = uf:getroot(b)print(ra == rb and "Yes" or "No")end