我正在用 Lua 编写一个非常简单的程序,以了解有关遗传编程的更多信息。下面是一个 mutate 函数,用于转到nodeNum
树 ( ) 中的编号节点 ( pop
) 以及:将 add 与 sub 交换(反之亦然)或用随机数 1-100 替换节点。
local count = 0
function mutate(pop, nodeNum)
for k,v in ipairs(pop) do
if type(v) == "table" then
mutate(v, nodeNum)
else
count = count + 1
end
if count == nodeNum and k == 1 then -- correct node
if type(v) == "function" then
if v == add then
pop[k] = sub
else
pop[k] = add
end
else
pop[k] = math.random(100)
end
end
end
end
我的问题是count
. 调用这个函数很尴尬,因为count
每次都必须重置:
-- mutate the first 3 elements in program tree t
mutate(t,1)
count = 0
mutate(t, 2)
count = 0
mutate(t, 3)
我尝试过使用do ... end
类似的变化:
do
local count
function mutate(pop, nodeNum)
if not count then
count = 0
...
end
我也尝试给 mutate 一个额外的参数mutate(pop, nodeNum, count)
并调用它,mutate(t, 1, 0)
但我无法让任何一种方法正常工作。
我可能遗漏了一些非常明显的东西,有人能看到更优雅的解决方案吗?