Lua学习十一----------Lua迭代器

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› LUA›LUA教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

Lua迭代器

　　- 迭代器（iterator）是一种对象，它能够用来遍历标准模板库容器中的部分或全部元素，每个迭代器对象代表容器中的确定的地址

　　- Lua迭代器是一种支持指针类型的结构，它可以遍历集合的每一个元素

泛型for迭代器

　　- 泛型for的执行过程：　

　　　　1）初始化，计算in后面表达式的值，表达式应该返回泛型for需要的三个值：迭代函数、状态常量、控制变量；个数不足用nil补充，多余忽略

　　　　2）将状态常量和控制变量作为参数调用迭代函数（对于for结构来说，状态常量没有作用，仅仅在初始化时获取它的值并传递给迭代变量）

　　　　3）将迭代函数返回的值赋给变量列表

　　　　4）如果返回的第一个值为nil，循环结束；否则执行循环体

　　　　5）回到第二步，再次调用迭代函数

　　- 默认的迭代函数

　　　　1）pairs：迭代table，可以遍历表中的所有key，可以返回nil

　　　　2）ipairs：迭代数组，从下标1开始升序遍历，遇到nil则退出

无状态的迭代器

　　- 无状态的迭代器是指不保留任何状态的迭代器，因此在循环中可以利用无状态迭代器避免创建闭包发费的额外代价

　　- 每一次迭代，迭代函数都是用两个变量（状态常量和控制变量）的值作为参数被调用

　　- ipairs是最简单的典型例子

多状态的迭代器

　　- 很多情况下，迭代器需要保存多个状态信息而不是简单的状态常量和控制变量。

　　- 方法一是使用闭包

　　- 方法二是将所有的状态信息封装到table中，将table作为迭代器的状态常量

实例

print("泛型for迭代器")

array = {
	[1] = "V1",
	[3] = "V2",
    key = "V3"
}

for k, v in pairs(array) do
	print(k .. " : " .. v)
end
print()
for k, v in ipairs(array) do
	print(k .. " : " .. v)
end

print("无状态的迭代器：实现数字n的平方")

function square(iteratorMaxCount, currentNumber)
	if iteratorMaxCount > currentNumber then
		currentNumber = currentNumber + 1
		return currentNumber, currentNumber * currentNumber
	end
end

for i, n in square, 3, 0 do
	print(i, n)
end

print("多状态的迭代器：计算集合大小并输出各个元素")

array = {"Lua", "Tutorial"}

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

参考：http://www.runoob.com/lua/lua-iterators.html