lua中table如何安全移除元素

  • MAY 20, 2014 | 4 COMMENTS

    在Lua中,table如何安全的移除元素这点挺重要,因为如果不小心,会没有正确的移除,造成内存泄漏。

    引子

    比如有些朋友常常这么做,大家看有啥问题

    将test表中的偶数移除掉
    local test = { 2, 3, 4, 8, 9, 100, 20, 13, 15, 7, 11}
    for i, v in ipairs( test ) do
        if v % 2 == 0 then
            table.remove(test, i)
        end
    end
    
    for i, v in ipairs( test ) do
        print(i .. "====" .. v)
    end
    

    打印结果:

    1====3
    2====8
    3====9
    4====20
    5====13
    6====15
    7====7
    8====11
    [Finished in 0.0s]
    

    有问题吧,20怎么还在?这就是在遍历中删除导致的。

    如何做呢?

    Let's get started!
    local test = { 'a','b','c','d','e','f','g','h','i','j','k','l','m','n','o','p' }
    local remove = { a = true, b = true, c = true, e = true, f = true, p = true }
    
    local function dump(table)
        for k, v in pairs( table ) do
            print(k)
            print(v)
            print("*********")
        end
    end
    

    说明:一般我们不在循环中删除,在循环中删除会造成一些错误。这是可以建立一个remove表用来标记将要删除的,如上面例子,把将要删除的标记为true

    方法1 从后往前删除

    for i = #test, 1, -1 do
        if remove[test[i]] then
            table.remove(test, i)
        end
    end
    
    dump(test)
    

    为什么不从前往后,朋友们可以测试,table.remove操作后,后面的元素会往前移位,这时候后续的删除索引对应的元素已经不是之前的索引对应的元素了。

    方法2 while删除

    local i = 1
    while i <= #test do
        if remove[test[i]] then
            table.remove(test, i)
        else
            i = i + 1
        end
    end
    
    dump(test)
    

    方法3 quick中提供的removeItem

    function table.removeItem(list, item, removeAll)
        local rmCount = 0
        for i = 1, #list do
            if list[i - rmCount] == item then
                table.remove(list, i - rmCount)
                if removeAll then
                    rmCount = rmCount + 1
                else
                    break
                end
            end
        end
    end
    
    for k, v in pairs( remove ) do
        table.removeItem(test, k)
    end
    
    dump(test)
    

    github地址

    • https://github.com/dabing1022/LuaAndLove2dLearningTest/blob/master/LuaTest09/test12.lua

    • https://github.com/dabing1022/LuaAndLove2dLearningTest/blob/master/LuaTest09/test13.lua


    原文地址: http://childhood.logdown.com/posts/200499/lua-table-how-to-safely-remove-an-element

MAY 20, 2014 | 4 COMMENTS

你可能感兴趣的:(lua中table如何安全移除元素)