Lua 迭代器
参考文章:
菜鸟教程。
https://cloud.tencent.com/developer/article/2203215
迭代器(iterator)是一种对象,它能够用来遍历标准模板库容器中的部分或全部元素,每个迭代器对象代表容器中的确定的地址。
在 Lua 中迭代器是一种支持指针类型的结构,它可以遍历集合的每一个元素。
泛型for迭代器
泛型 for 在自己内部保存迭代函数,实际上它保存三个值:迭代函数、状态常量、控制变量。
泛型 for 迭代器提供了集合的 key/value 对,语法格式如下:
for k, v in pairs(t) do
print(k, v)
end
上面代码中,k, v为变量列表;pairs(t)为表达式列表。
查看以下实例:
array = {"Google", "Runoob"}
for key,value in ipairs(array)
do
print(key, value)
end
输出结果:
以上实例中我们使用了 Lua 默认提供的迭代函数 ipairs。
下面我们看看泛型 for 的执行过程:
-
首先,初始化,计算 in 后面表达式的值,表达式应该返回泛型 for 需要的三个值:迭代函数、状态常量、控制变量;与多值赋值一样,如果表达式返回的结果个数不足三个会自动用 nil 补足,多出部分会被忽略。
TIPS:
for k,v in pairs(t) do print(k,v) end
其中k为控制变量,其值在循环过程中永远不会是nil,因为当其值为nil时循环就结束。
传入的t即为状态常量,其值不会在循环中改变,是不可变的。
而迭代函数可以通过下面无状态迭代器部分的知识了解。
-
第二,调用迭代函数的同时,将状态常量和控制变量作为参数传入(注意:对于 for 结构本身来说,状态常量的用处仅仅是在初始化时获取他的值并传递给迭代函数,具体的迭代控制是由控制变量及迭代函数一同完成)。
-
第三,将迭代函数返回的值(key,value)赋给变量列表。
-
第四,如果返回的key为nil循环结束,否则执行循环体。
-
第五,回到第二步再次调用迭代函数。
在Lua中我们常常使用函数来描述迭代器,每次调用该函数就返回集合的下一个元素。
我们可以自定义一个迭代器,使用Lua的for…in…结构来迭代,如下:
function values(t)
local i = 0
return function () i = i + 1; return t[i] end
end
-- 自定义迭代
t = {10,20,30}
iter = values(t)
while true do
local element = iter()
if element == nil then break end
print(element)
end
-- 使用for in迭代
for element in values(t) do
print(element)
end
输出结果:
Lua 的迭代器包含以下两种类型:
- 无状态的迭代器
- 多状态的迭代器
无状态的迭代器
无状态的迭代器是指不保留任何状态的迭代器,因此在循环中我们可以利用无状态迭代器避免创建闭包花费额外的代价。
每一次迭代,迭代函数都是用两个变量(状态常量和控制变量)的值作为参数被调用,一个无状态的迭代器只利用这两个值可以获取下一个元素。
这种无状态迭代器的典型的简单的例子是 ipairs,它遍历数组的每一个元素,元素的索引需要是数值。
以下实例我们使用了一个简单的函数来实现迭代器,实现 数字 n 的平方:
function square(iteratorMaxCount, currentNumber)
if currentNumber < iteratorMaxCount
then
currentNumber = currentNumber + 1
return currentNumber, currentNumber * currentNumber
end
end
for i,n in square,3,0
do
print(i,n)
end
输出结果:
迭代的状态包括被遍历的表(循环过程中不会改变的状态常量)和当前的索引下标(控制变量),ipairs 和迭代函数都很简单,我们在 Lua 中可以这样实现(内部的ipairs实现原理):
function iter (a, i)
i = i + 1
local v = a[i]
if v then
return i, v
end
end
function ipairs (a)
return iter, a, 0 -->迭代函数,状态常量,控制变量
end
array = {"Google", "Runoob"}
for key,value in ipairs(array)
do
print(key, value)
end
输出结果:
当 Lua 调用 ipairs(a) 开始循环时,他获取三个值:迭代函数 iter、状态常量 a、控制变量初始值 0;然后 Lua 调用 iter(a,0) 返回 1, a[1](除非 a[1]=nil);第二次迭代调用 iter(a,1) 返回 2, a[2]……直到第一个 nil 元素。
多状态的迭代器
很多情况下,迭代器需要保存多个状态信息而不是简单的状态常量和控制变量,最简单的方法是使用闭包,还有一种方法就是将所有的状态信息封装到 table 内,将 table 作为迭代器的状态常量,因为这种情况下可以将所有的信息存放在 table 内,所以迭代函数通常不需要第二个参数。
TIPS:闭包类似于一个匿名函数,迭代器本身就可以看作为一个闭包。
以下实例我们创建了自己的迭代器:
array = {"Google", "Runoob"}
function elementIterator (collection)
local index = 0
local count = #collection
-- 闭包函数
return function ()
index = index + 1
if index <= count
then
-- 返回迭代器的当前元素
return collection[index]
end
end
end
for element in elementIterator(array)
do
print(element)
end
输出结果:
以上实例中我们可以看到,elementIterator 内使用了闭包函数,实现计算集合大小并输出各个元素。