lua - 如何判断一个数组是否是另一个数组的子集？

Question

在 Python 中，我们可以使用 set 或 itertools 来查找另一个列表的一个列表的子集，我们如何在 Lua 中做同样的事情？

a = {1,2,3}
b = {2,3}

我如何检查 b 是 a 的子集？

Answer 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