在 Lua 中从数组中弹出第一个元素

Question

我在 Lua 中有一个数组 x 。我想设置 head = x[1] 和 rest = 数组的其余部分，以便 rest[1] = x[2]、rest[2] = x[3] 等。

我该怎么做？

（注意：我不关心原始数组是否发生变异。在 Javascript 中，我会执行 head = x.shift() ，并且 x 将包含剩余的元素。）

Answer 1

head = table.remove(x, 1)

“弹出" 有点用词不当，因为它意味着一种廉价的操作，并且删除表的第一个元素需要重新定位其余内容 - 因此在 JavaScript 和其他一些语言中称为“shift”。

Answer 2

您想要 table.remove：

local t = {1,2,3,4}
local head = table.remove(t,1)
print( head )
--> 1
print( #t )
--> 3
print( t[1] )
--> 2

---

正如 @daurnimator 指出的，这需要 Lua 运行时中数组的底层实现付出大量努力，移动所有表元素。如果您可以向后表示数组，调用数组中的最后一项 head，那么对 table.remove() 的调用将是一个廉价的 pop：

local t = {4,3,2,1}
local head = table.remove(t)
print(head)
--> 1
print( #t )
--> 3
print( t[#t] )
--> 2

或者，您可以选择将元素序列表示为链接列表。在这种情况下，从列表头部弹出一个项目也是一种廉价的操作（但将一个项目推到末尾则不是，除非您跟踪列表中的“尾部”）：

local setm,getm = setmetatable,getmetatable
local linkedlist=setm({__index={
  tail = function(l) while l.rest do l=l.rest end return l end, -- N.B. O(n)!
  push = function(l,v,t) t=l:tail() t.rest=setm({val=v},getm(l)) return t end,
  cram = function(l,v) return setm({val=v,rest=l},getm(l)) end,
  each = function(l,v)
    return function() if l then v,l=l.val,l.rest return v end end
  end
}},{ __call=function(lmeta,v,...)
  local head,tail=setm({val=v},lmeta) tail=head
  for i,v in ipairs{...} do tail=tail:push(v) end
  return head
end })

local numbers = linkedlist(1,2,3,4)
for n in numbers:each() do print(n) end
--> 1
--> 2
--> 3
--> 4

local head,rest = numbers.val, numbers.rest
print(head)
--> 1

for n in rest:each() do print(n) end
--> 2
--> 3
--> 4

local unrest = rest:cram('99')
for n in unrest:each() do print(n) end
--> 99
--> 2
--> 3
--> 4

特别注意，

local head,rest = numbers.val, numbers.rest

不会修改任何数据结构，但只是为您提供链中特定链接的rest句柄。

Answer 3

通常在 Lua 中将元素 x 插入序列的操作...

例如： S={a,b,c,d,e,f} 到 S={a,b,c,x,d,e,f ...

非常耗时，因为 d 必须移动到索引 5，e 移动到索引 6 等。

是否有其他 S 形式的序列，其中 S[a]=b，
S[b]=c、S[c]=d、S[d]=e 且 S[e]=f？这样，您所要做的就是输入：

S[c]=x
S[x]=d

并且，只需两次操作，x 就在 c 之后和 d 之前。