第十课 完整的Lua代码示例

马尔可夫链算法
根据原始文本中n个单词的序列来确定后面的单词，从而生成随机的文本。
程序先读取原始文本，并创建一个table。table的创建过程为：以每两个单词 为一个前缀，在table中创建一个列表，该列表记录了原始文件中所有位于该前缀之后的单词。当构建好这个table后，程序就利用它来生成随机文本。结果中的每个单词都来自于它在原始文本中的前缀，而具有相同前缀的单词出现在结果中的概率也与原始文本一样。最终会得到一串比较随机的文本。
将两个单词以空格相连，编码成一个前缀：
function prefix (w1, w2)

return w1 .. " " .. w2
end

使用字符串NOWORD（"\n"）来初始化一个前缀单词，并且标记 文本的结尾。例如，对于原始文本：
the more we try the more we do
table内容：
{
["\n" "\n"] = {"the"},
["\n the"] = {"more"},
["the more"] = {"we", "we"},
["more we"] = {"try", "do"},
["we try"] = {"the"},
["try the"] = {"more"},
["we do"] = {"\n"},
}
程序将这个table保存在变量statetab中。若要 向此table中的某个前缀列表插入一个新 单词，可以使用以下函数：
function insert (index, value)

local list = statetab[index]

if list == nil then

statetab[index] = {value} --没有就新加

else

list[#list + 1] = value --有就在列表末尾增加一个

end
end
辅助函数：
function allwords ()

local line = io.read()

local pos = 1

return function () --迭代器函数

while line do

local s, e = string.find(line, "%w+", pos)

if s then

pos = e + 1

return string.sub(line, s ,e)

else

line = io.read()

pos = 1

end

end

return nil

end
end

主程序：
local N = 2
local MAXGEN = 10000
local NOWORD = "\n"

--构建table
local w1, w2 = NOWORD, NOWORD
for w in allwords() do

insert(prefix(w1, w2), w)

w1 = w2; w2 = w
end
insert(prefix(w1, w2), NOWORD)

--生成文本
w1 = NOWORD; w2 = NOWORD
for i = 1, MAXGEN do

local list = statetab[prefix(w1, w2)]

--从列表中选择一个 随机项

local r = math.random(#list)

local nextword = list[r]

if nextword == NOWORD then return end

io.write(nextword, " ")

w1 = w2; w2 = nextword
end

输入：
i am a pretty boy and i have a beautiful pig
i love my pig forever and i hope her happiness as long as her life
if you also have a cherished person i hope you can stay with her as long as possible
可能的输出：
i am a pretty boy and i hope her happiness as long as possible