結果

問題 No.697 池の数はいくつか
ユーザー 👑 obakyan
提出日時 2019-05-12 11:50:18
言語 Lua
(LuaJit 2.1.1734355927)
結果
MLE  
実行時間 -
コード長 1,528 bytes
コンパイル時間 466 ms
コンパイル使用メモリ 6,948 KB
実行使用メモリ 395,976 KB
最終ジャッジ日時 2024-11-25 14:55:24
合計ジャッジ時間 16,626 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26 MLE * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

local ior = io.input()
local h, w = ior:read("*n", "*n")

local function getindex(i_h, i_w)
  return i_w + (i_h - 1) * w
end

local function createMap()
  local map = {}
  for i_h = 1, h do
    for i_w = 1, w do
      local idx = getindex(i_h, i_w)
      map[idx] = ior:read("*n")
    end
  end
  return map
end

local map = createMap()

local taskstate = {}
for i = 1, h * w do taskstate[i] = false end
local tasks = {}
local tasknum = 0
local done = 0
local tasklim = h * w

local function addtask(idx)
  if(not taskstate[idx]) then
    taskstate[idx] = true
    tasknum = tasknum + 1
    local taskidx = tasknum % tasklim
    if taskidx == 0 then taskidx = tasklim end
    tasks[taskidx] = idx
  end
end

local curpos = 1
local lakecount = 0
local function searchtask()
  for i = curpos, h * w do
    if(map[i] == 1) then
      lakecount = lakecount + 1
      curpos = i
      addtask(i)
      break
    end
  end
end
searchtask()
while(done < tasknum) do
  done = done + 1
  local taskidx = done % tasklim
  if(taskidx == 0) then taskidx = tasklim end
  local idx = tasks[taskidx]
  taskstate[idx] = false
  map[idx] = 2
  if(w < idx) then if(map[idx - w] == 1) then addtask(idx - w) end end
  if(idx <= (h - 1) * w) then if(map[idx + w] == 1) then addtask(idx + w) end end
  if(1 < w) then
    if(idx % w ~= 0) then if(map[idx + 1] == 1) then addtask(idx + 1) end end
    if(idx % w ~= 1) then if(map[idx - 1] == 1) then addtask(idx - 1) end end
  end
  if(done == tasknum) then
    searchtask()
  end
end
print(lakecount)
0