lua使用协同程序实现迭代器-笔记

function printResult( a )
    local str = ""
    for i=1,#a do
        str = str .. a[i] .. " "
    end
    print("--" .. str .. "\n")
end

function permgen(a, n)
    n = n or #a
    if n <= 1 then
        coroutine.yield(a) --完成一次排列 返回排列数组；挂起 保存迭代状态
    else
        for i=1,n do --每循环一次 生成一个a[i]所有位置变化的数组
            a[n], a[i] = a[i], a[n] --交换
            permgen(a, n - 1) -- 递归实现a[i]位置交换 每次循环交换至n-1
            a[n], a[i] = a[i], a[n] --还原 方便下个元素交换
        end
    end
end

function permutations(a)
    local co = coroutine.create(function () --创建协程
        permgen(a)
    end)
    return function ()
        local code, res = coroutine.resume(co) --每次调用迭代函数，即唤醒协同程序 产生一个值
        return res
    end
end

--迭代打印
for p in permutations{"a", "b", "c"} do
    printResult(p)
end