-
Notifications
You must be signed in to change notification settings - Fork 0
/
ex19.1.lua
78 lines (69 loc) · 1.67 KB
/
ex19.1.lua
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
local function allwords ()
local line = io.read()
-- current line
local pos = 1
-- current position in the line
return function ()
-- iterator function
while line do
-- repeat while there are lines
local w, e = string.match(line, "(%w+[,;.:]?)()", pos)
if w then
-- found a word?
pos = e
-- update next position
return w
-- return the word
else
line = io.read()
-- word not found; try next line
pos = 1
-- restart from first position
end
end
return nil
-- no more lines: end of traversal
end
end
local function prefix (words)
local format_string = string.format("%s%%s", string.rep("%s ", #words - 1))
return string.format(format_string, table.unpack(words))
end
local statetab = {}
local function insert (prefix, value)
local list = statetab[prefix]
if list == nil then
statetab[prefix] = {value}
else
list[#list + 1] = value
end
end
io.input("sample.txt")
local MAX_WORDS = 200
local NOWORD = "\n"
local SIZE = 4
local words = {}
for _ = 1, SIZE do
table.insert(words, NOWORD)
end
for next_word in allwords() do
insert(prefix(words), next_word)
table.remove(words, 1)
table.insert(words, next_word)
end
insert(prefix(words), NOWORD)
words = {}
for _ = 1, SIZE do
table.insert(words, NOWORD)
end
for _ = 1, MAX_WORDS do
local list = statetab[prefix(words)]
local r = math.random(#list)
local next_word = list[r]
if next_word == NOWORD then
return
end
io.write(next_word, " ")
table.remove(words, 1)
table.insert(words, next_word)
end