Как узнать, является ли массив подмножеством другого массива?

В 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
Другие вопросы по тегам