如何检查表是否包含Lua中的元素？

Question

有没有一种方法可以检查一个表是否包含一个值？我有自己的（天真的）功能，但我想知道是否存在“官方”的东西？或者更高效的方法...

function table.contains(table, element)
  for _, value in pairs(table) do
    if value == element then
      return true
    end
  end
  return false
end

顺便说一句，我使用这个函数的主要原因是使用表格作为集合，即没有重复的元素。还有什么我可以用的吗？

Answer 1

您可以将值作为表的键。例如：

function addToSet(set, key)
    set[key] = true
end

function removeFromSet(set, key)
    set[key] = nil
end

function setContains(set, key)
    return set[key] ~= nil
end

还有一个功能更全面的例子here。

Answer 2

鉴于您的表述，您的功能已尽其所能。当然，正如其他人所指出的（以及在 Lua 之前的语言中所实践的），您真正问题的解决方案是改变表示。当您有表格并且想要集合时，您可以使用 set 元素作为键并使用 true 作为值将表格转换为集合。 +1 给 interjay。

Answer 3

我知道这是一篇旧帖子，但我想为后代添加一些内容。 处理您遇到的问题的简单方法是创建另一个表，对键值。

即。您有 2 个具有相同值的表，一个指向一个方向，一个指向另一个。

function addValue(key, value)
    if (value == nil) then
        removeKey(key)
        return
    end
    _primaryTable[key] = value
    _secodaryTable[value] = key
end

function removeKey(key)
    local value = _primaryTable[key]
    if (value == nil) then
        return
    end
    _primaryTable[key] = nil
    _secondaryTable[value] = nil
end

function getValue(key)
    return _primaryTable[key]
end

function containsValue(value)
    return _secondaryTable[value] ~= nil
end

然后您可以查询新表以查看它是否具有键“元素”。这避免了遍历另一个表的每个值的需要。

如果事实证明您实际上不能将“元素”用作键，例如因为它不是字符串，那么例如在其上添加校验和或tostring，然后将其用作键.

为什么要这样做？如果您的表非常大，则遍历每个元素的时间会很长，从而使您无法经常这样做。额外的内存开销相对较小，因为它将存储 2 个指向同一个对象的指针，而不是同一个对象的 2 个副本。 如果您的表非常小，那么它的影响会小得多，事实上，迭代甚至可能比另一个地图查找更快。

然而，问题的措辞强烈暗示您有大量的项目需要处理。

Answer 4

我想不出另一种比较值的方法，但是如果您使用集合的元素作为键，您可以将值设置为 nil 以外的任何值。然后您无需搜索整个表即可快速查找。

Answer 5

-- in some helper module
function utils_Set(list)
    local set = {}
    for _, l in ipairs(list) do set[l] = true end
    return set
end

-- your table here
long_table = { "v1", "v2", "v1000"}

-- Consult some value
_set = utils_Set(long_table)
if _set["v1"] then print("yes!") end