結果

問題 No.2202 贅沢てりたまチキン
ユーザー 👑 obakyanobakyan
提出日時 2023-02-06 23:18:45
言語 Lua
(LuaJit 2.1.1734355927)
結果
AC  
実行時間 278 ms / 2,000 ms
コード長 1,322 bytes
コンパイル時間 136 ms
コンパイル使用メモリ 5,120 KB
実行使用メモリ 44,316 KB
最終ジャッジ日時 2024-07-05 00:46:37
合計ジャッジ時間 4,300 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

local n, m = io.read("*n", "*n")
local ea, eb = {}, {}
local edge = {}
local len = {}
local asked = {}
for i = 1, n do
  edge[i] = {}
  len[i] = 0
  asked[i] = false
end
for i = 1, m do
  ea[i], eb[i] = io.read("*n", "*n")
  table.insert(edge[ea[i]], eb[i])
  table.insert(edge[eb[i]], ea[i])
end
local parent = {}
for i = 1, n 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

for i = 1, m do
  local a, b = ea[i], eb[i]
  local ra, rb = uf_findroot(a), uf_findroot(b)
  parent[b], parent[rb] = ra, ra
end
local root = {}
for i = 1, n do
  local r = uf_findroot(i)
  root[r] = true
end
for r, _u in pairs(root) do
  asked[r] = true
  local tasks = {r}
  local done = 0
  local f = false
  while done < #tasks do
    done = done + 1
    local src = tasks[done]
    for i = 1, #edge[src] do
      local dst = edge[src][i]
      if not asked[dst] then
        asked[dst] = true
        len[dst] = len[src] + 1
        table.insert(tasks, dst)
      else
        if (len[src] + len[dst]) % 2 == 0 then
          f = true
        end
      end
    end
  end
  if not f then print("No") os.exit() end
end
print("Yes")
0