lua中table如何安全移除元素?

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