Как узнать, является ли массив подмножеством другого массива?
В Python мы можем использовать set или itertools, чтобы найти подмножество одного списка другого списка, как мы делаем то же самое в Lua?
a = {1,2,3}
b = {2,3}
Как я могу проверить, что b является подмножеством a?
1 ответ
Решение
Наборы могут быть реализованы в Lua с использованием таблиц в качестве поиска для тестирования членства ( как это сделано в Программирование в Lua). Ключи в таблице являются элементами набора, а значения либо true
если элемент принадлежит набору или nil
иначе.
a = {[1]=true, [2]=true, [3]=true}
b = {[2]=true, [3]=true}
-- Or create a constructor
function set(list)
local t = {}
for _, item in pairs(list) do
t[item] = true
end
return t
end
a = set{1, 2, 3}
b = set{2, 3}
Написание операций над множествами также просто в этой форме ( как здесь).
function subset(a, b)
for el, _ in pairs(a) do
if not b[el] then
return false
end
end
return true
end
print(subset(b, a)) -- true
print(subset(set{2, 1}, set{2, 2, 3, 1})) -- true
a[1] = nil -- remove 1 from a
print(subset(a, b)) -- true
Если a
а также b
должен оставаться в форме массива, тогда подмножество может быть реализовано так:
function arraysubset(a, b)
local s = set(b)
for _, el in pairs(a) -- changed to iterate over values of the table
if not s[el] then
return false
end
end
return true
end