【Lua】迭代器初学

和C++类似，Lua也有迭代器，如泛型for。泛型for需要三个值：迭代函数、状态常量、控制变量

一个典型的泛型for如下：

for k, v in pairs(t) do
    print(k, v)
end

泛型for的执行过程：
首先，初始化，计算 in 后面表达式的值，表达式应该返回泛型for需要的三个值：迭代函数、状态常量、控制变量。（并不保证一定会返回三个值，有时只会传递迭代函数，其他值就为nil）
第二，将状态常量和控制变量作为参数调用迭代函数（对于for结构来说，状态常量并没有用到，仅仅在初始化时获取它的值并传递给迭代函数）。
第三，将迭代函数返回的值赋给for的变量列表。
第四，如果返回的第一个值为nil，则循环结束，否则执行循环体。
第五，回到第二步再次调用迭代函数

无状态的迭代器：

无状态的迭代器是指不保留任何状态的迭代器，因此在循环中我们可以利用无状态迭代器避免创建闭包花费额外的代价。每一次迭代，迭代函数都是用两个变量（状态常量和控制变量）的值作为参数被调用，一个无状态的迭代器只利用这两个值可以获取下一个元素。这种无状态迭代器的典型的简单的例子是ipairs，他遍历数组的每一个元素。

一个简易的无状态迭代器实现如下：

function iter (a, i)
    print("iter")
    i = i + 1
    if a[i] then
	local v = a[i] ^ 2
	return i, v
    end
end
 
function ipairs (a)
    print("ipairs")
    return iter, a, 0
end

local tab = {1, 2}
for k, v in ipairs(tab) do
	print(k, v)
end

运行结果如下：

ipairs
iter
1	1.0
iter
2	4.0
iter

可以发现 ipairs 函数只运行了一次，iter函数运行多次。

至于

iter, a, 0

就是返回的迭代函数、状态常量以及控制变量三个参数。

在以上的写法中，lua调用ipairs(a)开始循环，它先获取三个值：迭代函数 iter、状态常量 a、控制变量初始值 0，然后 lua调用 iter(a,0) 返回1, a[1]（除非a[1]=nil），第二次迭代调用iter(a,1)返回2,a[2]……直到第一个nil元素，此时 iter 不会调用 return 函数，for语句中 k, v 的值就都会是 nil，随即 for 语句结束。

这里有一个疑问，为什么第二次会调用 iter(a, 1)，第三次 iter(a, 2)？还望指教貌似是通过 iter 函数第一个返回值进行控制变量的迭代。

多状态的迭代器：

很多情况下，迭代器需要保存多个状态信息而不是简单的状态常量和控制变量，最简单的方法是使用闭包，还有一种方法就是将所有的状态信息封装到table内，将table作为迭代器的状态常量，因为这种情况下可以将所有的信息存放在table内，所以迭代函数通常不需要第二个参数。

一个简易的多状态迭代器实现如下：

local tab = {1, 2}
function iter2 (a)
	local index = 0
	local len = #a
	
	-- 闭包函数
	return function()
		index = index + 1
		if a[index] then
			return index, a[index] ^ 2
		end
	end
end

for k, v in iter2(tab) do
	print(k, v)
end

输出：

1	1.0
2	4.0

可以发现相比无状态迭代器，多状态迭代器使用了闭包，也即使用了内嵌函数，并将控制变量变成一个upvalue。