Create deque.lua
This commit is contained in:
parent
7ceebbb411
commit
0993b03088
121
lib/utils/deque.lua
Normal file
121
lib/utils/deque.lua
Normal file
@ -0,0 +1,121 @@
|
||||
---@class Deque
|
||||
---@field private first integer
|
||||
---@field private last integer
|
||||
---@field private _data table<integer, any>
|
||||
local Deque = {}
|
||||
Deque.__index = Deque
|
||||
|
||||
---Создать пустую очередь
|
||||
---@return Deque
|
||||
local function new()
|
||||
---@type Deque
|
||||
local self = setmetatable({
|
||||
first = 1,
|
||||
last = 0,
|
||||
_data = {},
|
||||
}, Deque)
|
||||
return self
|
||||
end
|
||||
|
||||
---Количество элементов
|
||||
---@return integer
|
||||
function Deque:size()
|
||||
return self.last - self.first + 1
|
||||
end
|
||||
|
||||
---Пуста ли очередь
|
||||
---@return boolean
|
||||
function Deque:is_empty()
|
||||
return self.first > self.last
|
||||
end
|
||||
|
||||
---Добавить в начало
|
||||
---@param value any
|
||||
---@return Deque self
|
||||
function Deque:push_front(value)
|
||||
self.first = self.first - 1
|
||||
self._data[self.first] = value
|
||||
return self
|
||||
end
|
||||
|
||||
---Добавить в конец
|
||||
---@param value any
|
||||
---@return Deque self
|
||||
function Deque:push_back(value)
|
||||
self.last = self.last + 1
|
||||
self._data[self.last] = value
|
||||
return self
|
||||
end
|
||||
|
||||
---Забрать из начала
|
||||
---@return any
|
||||
function Deque:pop_front()
|
||||
if self.first > self.last then return nil end
|
||||
local value = self._data[self.first]
|
||||
self._data[self.first] = nil
|
||||
self.first = self.first + 1
|
||||
return value
|
||||
end
|
||||
|
||||
---Забрать из конца
|
||||
---@return any
|
||||
function Deque:pop_back()
|
||||
if self.first > self.last then return nil end
|
||||
local value = self._data[self.last]
|
||||
self._data[self.last] = nil
|
||||
self.last = self.last - 1
|
||||
return value
|
||||
end
|
||||
|
||||
---Подсмотреть начало
|
||||
---@return any
|
||||
function Deque:peek_front()
|
||||
if self.first > self.last then return nil end
|
||||
return self._data[self.first]
|
||||
end
|
||||
|
||||
---Подсмотреть конец
|
||||
---@generic T
|
||||
---@return T|nil
|
||||
function Deque:peek_back()
|
||||
if self.first > self.last then return nil end
|
||||
return self._data[self.last]
|
||||
end
|
||||
|
||||
---Очистить очередь
|
||||
function Deque:clear()
|
||||
-- Полная очистка таблицы для GC
|
||||
for i = self.first, self.last do
|
||||
self._data[i] = nil
|
||||
end
|
||||
self.first, self.last = 1, 0
|
||||
end
|
||||
|
||||
---Итератор по значениям слева направо
|
||||
---@return fun():any
|
||||
function Deque:values()
|
||||
local i = self.first
|
||||
return function()
|
||||
if i <= self.last then
|
||||
local v = self._data[i]
|
||||
i = i + 1
|
||||
return v
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
---Преобразовать в массив
|
||||
---@return any[]
|
||||
function Deque:to_array()
|
||||
local n = self:size()
|
||||
local arr = {}
|
||||
if n <= 0 then return arr end
|
||||
local j = 1
|
||||
for i = self.first, self.last do
|
||||
arr[j] = self._data[i]
|
||||
j = j + 1
|
||||
end
|
||||
return arr
|
||||
end
|
||||
|
||||
return { new = new }
|
||||
Loading…
x
Reference in New Issue
Block a user