Question

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

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

如何检查b是否为？

的子集

Answer 1

使用表作为成员资格测试的查找（as done in Programming in 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}

在这种形式下编写集合操作也很简单（as here）。

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

如何查找数组是否是另一个数组的子集？

1 个答案: