【Lua 入门基础篇（六）】数组&迭代器

一、数组

1. 一维数组

数组，就是相同数据类型的元素按一定顺序排列的集合，可以是一维数组和多维数组。

Lua 数组的索引键值可以使用整数表示，数组的大小不是固定的。

a = {'lua', 'js', 'c++', 'python', 'java', 'docker'}

for i = 1, #a do
    print(a[i])
end
1
2
3
4
5

lua
js
c++
python
java
docker
1
2
3
4
5
6

在 Lua 索引值是以 1 为起始，但你也可以指定 0 开始。如果指定的索引没有值则返回 nil。

此外我们还可以用负数作为数组索引值：

a = {}

for i= -2, 2 do
   a[i] = i * 2
end

for i = -2, 2 do
   print(a[i])
end
1
2
3
4
5
6
7
8
9

-4
-2
0
2
4
1
2
3
4
5

---

2. 多维数组

多维数组即数组中包含数组或一维数组的索引键对应一个数组。

a = {}
for i = 1, 3 do
   a[i] = {}
      for j= 1, 3 do
         a[i][j] = i * j
      end
end

for i = 1, 3 do
   for j = 1, 3 do
      print(a[i][j])
   end
end
1
2
3
4
5
6
7
8
9
10
11
12
13

1
2
3
2
4
6
3
6
9
1
2
3
4
5
6
7
8
9

---

二、迭代器

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

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

1. 泛型 for 迭代器

泛型 for 在自己内部保存迭代函数，实际上它保存三个值：迭代函数、状态常量、控制变量。

泛型 for 迭代器提供了集合的 key/value 对，语法格式如下：

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

k, v为变量列表；pairs(t)为表达式列表。

泛型 for 的执行过程：

• 首先，初始化，计算 in 后面表达式的值，表达式应该返回泛型 for 需要的三个值：迭代函数、状态常量、控制变量；与多值赋值一样，如果表达式返回的结果个数不足三个会自动用 nil 补足，多出部分会被忽略。
• 第二，将状态常量和控制变量作为参数调用迭代函数（注意：对于 for 结构来说，状态常量没有用处，仅仅在初始化时获取他的值并传递给迭代函数）。
• 第三，将迭代函数返回的值赋给变量列表。
• 第四，如果返回的第一个值为nil循环结束，否则执行循环体。
• 第五，回到第二步再次调用迭代函数

在Lua中我们常常使用函数来描述迭代器，每次调用该函数就返回集合的下一个元素。

Lua 的迭代器包含以下两种类型：

• 无状态的迭代器
• 多状态的迭代器

---

2. 无状态的迭代器

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

每一次迭代，迭代函数都是用两个变量（状态常量和控制变量）的值作为参数被调用，一个无状态的迭代器只利用这两个值可以获取下一个元素。

下面用一个简单的函数来实现迭代器：实现数字 n 的平方：

function square(iteratorMax, cur)
    if cur < iteratorMax then
        cur = cur + 1
        return cur, cur * cur
    end
end

for i, n in square, 3, 0 do
    print(i, n)
end
1
2
3
4
5
6
7
8
9
10

运行结果：

1    1
2    4
3    9
1
2
3

迭代的状态包括被遍历的表（循环过程中不会改变的状态常量）和当前的索引下标（控制变量）。

类似实现ipairs：

function iterator(a, i)
    i = i + 1
    local v = a[i]
    if v then return i, v end
end

function ipairs(a)
    return iterator, a, 0
end

a = {"1", "2", "3", "4"}
for k, v in ipairs(a) do
    print(k, v)
end
1
2
3
4
5
6
7
8
9
10
11
12
13
14

运行结果：

1	1
2	2
3	3
4	4
1
2
3
4

a = {}
for i = -2, 2 do
   a[i] = i
end
for k, v in ipairs(a) do
    print(k, v)
end
1
2
3
4
5
6
7

运行结果：

1	1
2	2
1
2

当 Lua 调用 ipairs(a) 开始循环时，他获取三个值：迭代函数 iter、状态常量 a、控制变量初始值 0；

然后 Lua 调用 iter(a,0) ，返回 1, a[1]（除非 a[1]=nil）；

第二次迭代调用 iter(a,1) ，返回 2, a[2]……直到第一个 nil 元素。

---

3. 多状态的迭代器

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

function elementIterator(collection)
    local index = 0
    local len = #collection
    return function()
        index = index + 1
        if index <= len then
            return collection[index]
        end
    end
end

a = {'python', "lua", "typescript"}
for element in elementIterator(a) do
    print(element)
end
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

python
lua
typescript
1
2
3

elementIterator 内使用了闭包函数，实现计算集合大小并输出各个元素。

---