vector - 从Lua中的2个向量中获取公共元素

Question

我有 2 个向量，比如说v1 = (a,b,c,d,e)andv2 = (e,h,t,b,w)我想得到第三个向量，其中包含 and 的共同元素v1，v2在这种情况下v3 = (e,b)

我已经看到这已经在 c++ 中被问过了，但我在 Lua 中看不到它。

Answer 1

我假设在说向量时，您的意思是v1并且v2都是类似数组的表，如下所示：

local v1 = {1,2,4,8,16}
local v2 = {2,3,4,5}

然后你可以这样做：

local v3 = {}

for k1,v1 in pairs(v1) do
    for k2,v2 in pairs(v2) do
        if v1 == v2 then
            v3[#v3 + 1] = v1
        end
    end
end

Answer 2

以防万一你想与更大的向量（数组/集合）相交，这里是一个未测量的过早优化的尝试

local v1 = {1,2,4,8,16}
local v2 = {2,3,4,5}

local function intersect(t1,t2)

        local function make_lookup(t)
            local res={}
            for _,v in ipairs(t) do
                res[v]=true
            end
            return res
        end

        local smaller,larger
        if (#t1>#t2) then
            larger=t1
            smaller=t2
        else
            larger=t2
            smaller=t1
        end

        local lookup=make_lookup(smaller)

        local res={}
        for _,v in ipairs(larger) do
            if lookup[v] then
                res[#res+1]=v
            end
        end
        return res
end

local v1v2_intersected=intersect(v1,v2)

内部查找表由两者中较小的数组表组成。在“生产”中，您可能还想检查您是否正在处理正确的输入值

Answer 3

您可以通过v1构建其值的查找表，然后通过v2以查找公共值并将它们添加到输出列表中：

function findcommon(a1, a2)
    local in_a1 = {} -- lookup table for a1's values
    for k,v in pairs(a1) do in_a1[v] = true end

    local common = {}
    local n = 0
    for k,v in pairs(a2) do 
        if in_a1[v] then 
            in_a1[v] = false
            n = n + 1
            common[n] = v
        end
    end

    return common
end

鉴于你的例子：

v1 = {'a','b','c','d','e'} 
v2 = {'e','h','t','b','w'}
common = findcommon(v1,v2)

print('('..table.concat(common,',')..')')  --> (e,b)

Answer 4

与于浩的解决方案不同，下面的代码运行在线性时间，但它确实需要额外的内存，除非你可以忍受v3而不是坚持v4。不过，时间差异可能只在大输入向量时才明显。

local v1={1,2,4,8,16}
local v2={2,3,4,5}
local v3={}
for k,v in ipairs(v1) do v3[v]=false end
for k,v in ipairs(v2) do if v3[v]==false then v3[v]=true end end
for k,v in  pairs(v3) do if v3[k]==false then v3[k]=nil end end
local v4={}
local n=0
for k,v in  pairs(v3) do n=n+1; v4[n]=k end
for k,v in ipairs(v4) do print(k,v) end