欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Lua中的迭代器浅析

程序员文章站 2022-03-07 08:55:53
今天学习的内容还蛮有意思的,让我兴奋了一下~ 1.迭代器 什么是迭代器?别傻了,我最讨厌的就是名词解释了,反正就是用来遍历集合的一种方式。 比如,我们最常用的pair...

今天学习的内容还蛮有意思的,让我兴奋了一下~

1.迭代器

什么是迭代器?别傻了,我最讨厌的就是名词解释了,反正就是用来遍历集合的一种方式。
比如,我们最常用的pairs,如下代码:

复制代码 代码如下:

    local t = {"fdsd", "445"};
  
    for k, v in pairs(t) do
        print("k=" .. k .. ", v=" .. v);
    end

 
这是一次遍历table的操作,然后打印出table的key值和value值。
输出结果如下:
复制代码 代码如下:

[lua-print] k=1, v=fdsd
[lua-print] k=2, v=445

2.自己写一个迭代器

要想了解迭代器,那还是要自己写一个才行,迭代器没有什么神奇的地方,它很简单。如这样一个函数:

复制代码 代码如下:

function diedaiqi(t)
    local i = 0;
    return function()
        i = i + 1;           
        return t[i];
    end
end

这函数故名思议,叫做迭代器,英文不好的也没关系,知道是这意思就好了,呵呵(小若:英文你个头啊!分明是拼音啊!)
 
有没有发现这diedaiqi函数有点特别?没错,它就是之前的文章提到过的“闭合函数”,正是利用了闭合函数的特性来实现迭代功能的。
来看看如何使用这个迭代器吧:

复制代码 代码如下:

    local iter = diedaiqi(t);
    while true do
        local value = iter();
        if value == nil then
            break;
        end
      
        print(value);
    end

因为每一次调用diedaiqi函数,就会产生一个新的闭合函数,所以我们要用一个iter变量保存这个闭合函数,避免重复创建。

如果你对闭合函数已经很生疏了,可以看看我之前的这篇文章:【笨木头lua专栏】基础补充03:闭合函数、非全局函数与函数的尾调用

接着,只要循环调用iter闭合函数即可,因为闭合函数的特点,i变量是会一直增加的,所以每次调用iter函数,返回的都是下一个table元素。
最终输出结果如下:

复制代码 代码如下:

[lua-print] fdsd
[lua-print] 445

3.更简洁的迭代器调用

刚刚调用迭代器的方式也太粗暴了,这么长一片代码,不太合理。
所以,我们又有了偷懒的方式——使用for循环调用迭代器。
 
直接看代码,刚刚的迭代器可以这么调用:

复制代码 代码如下:

    local t = {"fdsd", "445"};
    for value in diedaiqi(t) do
        print(value);
    end

 
这里大家可能会有一个疑问,每一次的循环,都会调用一次diedaiqi函数,那不就会产生很多个闭合函数?那i的值不就每次都是0?

答案是:不会的。

因为for循环只会调用一次diedaiqi函数,然后把它的返回值保存起来。

4.结束

这篇的介绍似乎是异常地简短,其实不是的,还有下篇,因为我怕接下来要说的东西比较多,导致文章太长。

所以,还是分开来介绍吧,下一篇,我们来深入了解一下为什么for循环可以这么方便地处理迭代器。