Question

我有2个向量，让我们说v1 = (a,b,c,d,e)和v2 = (e,h,t,b,w)我希望获得包含v1和v2的公共元素的第三个向量，在这种情况下{ {1}}

我已经看到这已经在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

与Yu Hao的解决方案不同，下面的代码以线性时间运行，但它需要额外的内存，除非您可以使用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

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

如果你想要交叉较大的矢量（数组/集合），这里是一个未经测量的过早优化的尝试

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)

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

从Lua中的2个向量中获取共同元素

4 个答案: