【Lua】for循环基础问题

Question

Lua for

下列**正确**代码实现功能：

计算1至5区间每个数的平方。

function square(max,n)
    if n < max
    then
        n = n + 1
        return n, n * n
    end
end

for i,n in square,5,0
do
    print(i,n)
end

输出：

1   1
2   4
3   9
4   16
5   25

问题现象：

在square函数中，增加else分支，返回1，1。如下：

function square(max,n)
    if n < max
    then
        n = n + 1
        return n, n * n
    else    -- 增加分支
        return 1,1
    end
end

for i,n in square,5,0
do
    print(i,n)
end

输出：

1   1
2   4
3   9
4   16
5   25
1   1
2   4
3   9
4   16
5   25
1   1
2   4
....

问题：

我认为输出应该是：

1   1
2   4
3   9
4   16
5   25
1   1
1   1
1   1
....

我个人认为，当循环进行到第6次以后，square函数n参数等于5，没有条件使其增加，一直返回1，1。

第7次为什么n又变成2了，哪个条件改变了n的值?

Answer 1

lua的for in详解用法，参考博客
https://blog.csdn.net/liuyuan185442111/article/details/54144348
总结一下：
《Lua程序设计》(第4版)给出了的解答，
for var_1,...,var_n in explist do block end
等价于

do
    local _f,_s,_var = explist
    while true do
        local var_1,...,var_n = _f(_s, _var)
        _var = var_1
        if _var == nil then break end
        block
    end
end

将题主所示正确代码套入，当n>=max时候，没有返回值，即此时_var = var_1 = nil，所以循环中断
再将题主代码套入，当n>=max时候，返回1,此时不会影响第一个参数5（即代码中的_s），只会将第二个参数进行赋值 ，即此时_var = var_1 = 1，所以此时输出从1开始，且不会跳出循环。
所以可以将题主的else中的return理解为：
假设此处return a,b，走到当前行时，将当前结果赋值i=a,n=b，且下一次执行square(5,a)