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

这里有一个功能更齐全的示例。

Answer 2

给定你的表示，你的函数是尽可能高效的。当然，正如其他人所指出的（以及早于 Lua 的语言中的实践），真正问题的解决方案是改变表示。当您有表并且需要集合时，可以通过使用 set 元素作为键并使用 true 作为值将表转换为集合。 +1 为 interjay。

Answer 3

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

IE。您有两张具有相同值的表，一张指向一个方向，一张指向另一个方向。

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

然后，您可以查询新表以查看它是否具有键“element”。这样就无需遍历另一个表的每个值。

如果事实证明您实际上无法使用“元素”作为键，因为它不是字符串，那么可以在其上添加校验和或tostring，然后将其用作关键。

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

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

Answer 4

-- 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

Answer 5

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