Lua:k维点集

我希望确保我随机生成的 kD 点不会相同。为了有效地进行测试,我想使用一个 set(至少在 C++ 中是这样做的)。

我知道在 Lua 中可以通过将对象插入值为任何非 nil 键(通常只是 true 或整数用于多集)的表中来将表用作 set。

但问题是,在 Lua 中,默认情况下仅按其地址比较表,即使所有向量分量(在我的情况下)都相等,这也是不同的。

所以我想为我的点实现 equal 和/或 less than 元方法。但这没用(见下面的代码)。有人可以帮忙吗?

local k = 3
local mt = {}

mt.__eq = function( a, b )
    for dim = 1, k do
        if a[dim] ~= b[dim] then return false end
    end
    return true
end

mt.__lt = function( a, b )
    for dim = 1, k do
        if     a[dim] < b[dim] then
            return true
        elseif a[dim] > b[dim] then
            return false
        end
    end
    return false -- equal
end

local set = {}

local p1 = { 1, 2, 3 }
setmetatable( p1, mt )
set[p1] = true

local p2 = { 1, 2, 3 }
setmetatable( p2, mt )
set[p2] = true -- should just overwrite the old value

print( "p1 == p2 --> "..tostring( p1 == p2 ) )
print( "p1 <  p2 --> "..tostring( p1 <  p2 ) )

local setSize = 0
for _, _ in pairs( set ) do
    setSize = setSize + 1
end
print( "Size of the set: "..setSize )
点赞
用户734069
用户734069

Lua 没有相应的机制来允许你所要实现的操作。也就是说,没有元方法的诡计可以让两个不同的用户数据或两个不同的表映射到同一张表的索引上。

你必须手动完成这个操作。你可以通过手动维护一个按照向量排序的列表来有效地编写自己的集合。table.sort 可以胜任这个工作,因为你已经有了一个适当的 __lt 元方法。

2015-12-22 20:23:22
用户5700094
用户5700094

如果有人感兴趣:我已经想出了另一种解决方法,对我有效。

可以使用一组 x 坐标,而不是仅将集合项与“true”或“1”相关联,而是将相应的值本身设为集合(y 坐标的集合),依此类推。

这导致了一种树状结构,其中每个叶子的路径代表一个点(坐标序列),具有相同第一个坐标的点具有共同的父节点。此外,所有点都是唯一的。

2015-12-29 08:13:57