Lua迭代器以及各种源函数的实现

范型for

范型for的格式如下所示:

Lua 复制代码
for <var-list> in <exp-list> do
    <body> 
end

var-list指变量名列表,可以为多个,exp-list指表达式列表,通常情况下只有一个值。可以更具体地写为另一种形式:

Lua 复制代码
for var_1, ..., var_n in explist(s) do block end

例如代码块中常见的for k,v in pairs(tb) do end或者for k,v in ipairs(tb) do end,都是这种范型形式,如若转换成while形式则如下:

Lua 复制代码
local func = explist
local s
while true do
    local var_1, ..., var_n = func(s)
    if var_1 == nil then
        break
    end
    block
end

源函数iter、ipairs、pairs实现

可以随意写一个简单的迭代器循环:

Lua 复制代码
function iter(a)
    local i = 0
    return function()
        i = i + 1
        return a[i]
    end
end

for n in iter({1,2,3}) do
    print(n) -- 1  2  3
end

相当于是以下代码结构:

Lua 复制代码
local iterator = iter({1,2,3})
local n = iterator()
while n do
    print(n)
    n = iterator()
end

ipairs函数只不过是上面iter内部的return a[i]改为return i,a[i]罢了。

pairs函数则是用到了next方法:

Lua 复制代码
function pairsMySelf(a)
    local key,value = next(a)
    return function()
        local oldKey,oldValue = key,value
        if value ~= nil then
            key,value = next(a,key)
        end
        return oldKey,oldValue
    end
end

local tb = {
    ["a"] = 1,
    ["b"] = 1,
    ["c"] = 1,
}
for k,v in pairsMySelf(tb) do
    print(k,v)
end
--[[print:
            a       1
            b       1
            c       1
]]--
相关推荐
我叫小白菜5 分钟前
【Java_EE】单例模式、阻塞队列、线程池、定时器
java·开发语言
狐凄12 分钟前
Python实例题:基于 Python 的简单聊天机器人
开发语言·python
weixin_446122461 小时前
JAVA内存区域划分
java·开发语言·redis
悦悦子a啊1 小时前
Python之--基本知识
开发语言·前端·python
QuantumStack2 小时前
【C++ 真题】P1104 生日
开发语言·c++·算法
whoarethenext2 小时前
使用 C++/OpenCV 和 MFCC 构建双重认证智能门禁系统
开发语言·c++·opencv·mfcc
代码的奴隶(艾伦·耶格尔)3 小时前
后端快捷代码
java·开发语言
Jay_5153 小时前
C++多态与虚函数详解:从入门到精通
开发语言·c++
路来了3 小时前
Python小工具之PDF合并
开发语言·windows·python
xiaolang_8616_wjl4 小时前
c++文字游戏_闯关打怪
开发语言·数据结构·c++·算法·c++20