lua4使用table实现其他数据结构，并介绍遍历方法

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› LUA›LUA教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

本文会以vector / map / set 这三种数据类型的角度来梳理 table 支持的不同遍历方式。

table as std::vector

一般，C/C++中的 array / vector （下文简称 vector）是没有 key。但是在 lua 中使用了 table 这种通用结构，就引入了 key 的问题。

在这里，把想用做 vector 的 table，做一个非常重要的约定：1 初始，key 连续。由于 table 的自由度很高，这个需要开发者自己约束。

---- 新建

t = {"A", "BB", "CCC"} -- 默认的key就是1初始且连续

-- 或者
t = {} -- 建立一个空的
t[1] = 1
t[2] = 2
t[3] = 3

-- 或者
t = {  
    [1] = 1,  
    [2] = 2,  
    [3] = 3,  -- 这里逗号可有可无
}

---- 模拟 pushback

vector = {}
function vector.pushback(vec, val)
  vec[#vec + 1] = val
end

v = {}
vector.pushback(v, 1)
vector.pushback(v, 2)
vector.pushback(v, 3)

---- 遍历，使用 ipairs，按照key从1开始，从小到大返回元素。

for key, value in ipairs(v) do
  print(key, value)
end

table as std::map 用key寻值，无重复。通用情况

---- 新建

m = {["A"] = 1, ["BB"] = 2, ["CCC"] = 3} 
m = {["A"] = 1, ["BB"] = "string2", 10 = 3}  -- key 和 value 的数据类型不必相同，自由伴随着控制的难度，一定程度上，key 和 value 推荐使用相同的类型。value 也可以是 table。handle 这种情况需要注意更多。

-- 或者
m = {} -- 建立一个空的
m[1] = 1
m[2] = 2
m["3"] = 3 -- 打印结果和 3 一样，可以通过 type 判断类型差别。详见后续例子

-- 或者
m = {  
    [1] = 1,  
    [2] = 2,  
    ["3"] = 3,  -- 这里逗号可有可无
}

---- 模拟 insert，不会重复插入

map = {}
function map.insert(map, key, val)
  map[key] = val
end

m = {["A"] = "str1"}
map.insert(m, 1, 25)
map.insert(m, "1", 5)
map.insert(m, 1, 2)
map.insert(m, 1, 25)

---- 查询元素，有就是1，没有就是0

function map.have(map, key)
  if map[key] == nil then
    return false
  else
    return true
  end
end

m = {["A"] = "str1"}
map.insert(m, 1, 25)
map.insert(m, "1", 5)
map.insert(m, 1, 2)
map.insert(m, 1, 25)

print(map.have(m, "A"))
print(map.have(m, A))

------- 结果 --------
true
false

----遍历，使用 pairs()，按照 key 的 harsh 值大小排序输出。注意，顺序这个概念在 map 中并不重要

for key, value in pairs(m) do
  print(key, type(key), value, type(value))
end

table as std::map 特殊情况：key 为数字，但不连续，希望从小到大遍历，或相反。

使用迭代器（自定义的）

-- from: program in Lua
function pairsByKeys(t)
    local a = {}
    for n in pairs(t) do 
      a[#a + 1] = n 
    end
    table.sort(a) -- 默认升序
　 -- table.sort(a, function(a1, a2) return a1 > a2 end) -- 降序
    local i = 0
    return function ()
        i = i + 1
        return a[i], t[a[i]]
    end
end

完整示例：

map = {}
function map.insert(map, key, val)
  map[key] = val
end

function map.have(map, key)
  if map[key] == nil then
    return false
  else
    return true
  end
end

-- from: program in Lua
function pairsByKeys(t)
    local a = {}
    for n in pairs(t) do 
      a[#a + 1] = n 
    end
    table.sort(a)
    local i = 0
    return function ()
        i = i + 1
        return a[i], t[a[i]]
    end
end

m = {}
map.insert(m, 1, 10)
map.insert(m, 3, 30)
map.insert(m, 9, 90)
map.insert(m, 5, 50)

print("pairsByKeys")
for key, value in pairsByKeys(m) do
  print(key, value)
end

print("pairs")
for key, value in pairs(m) do
  print(key, value)
end

----------- 结果 ------------
pairsByKeys
1	10
3	30
5	50
9	90
pairs
1	10
9	90
5	50
3	30

table as std::set 有序、不重复

这里的 map 和 set 的差别就是在插入的时候，需要对所有数据按照 key 排序，从小到大。在lua 里，map 已经没有重复的 key。

---- 新建

s = {}
s = {1,2,3}
s = {"A", "B", "C"} --> 字符串排序的规则需要自己定义

---- 模拟 insert，需要指定 value 的比较方式。set 是升序排列的，key 值不可改（注意赋值方式），所以需要指定 value 的比较方式。

set = {}
function set.insert(_s, val, _f) -- _f 就是需要自定义的比较函数，作为外部参数输入。详见后续
  _s[#_s + 1] = val
  if #_s > 1 then
    table.sort(_s, _f)
  end
end

---- 判断元素有无

set = {}
function set.have(s, key)
  if s[key] == nil then
    return false
  else
    return true
  end
end

---- 迭代所有元素使用：ipairs()

for key, value in ipairs(s) do
  print(key, value)
end

---- 完整示例　　

set = {}
function set.insert(_s, val, _f)
  _s[#_s + 1] = val
  if #_s > 1 then
    table.sort(_s, _f)
  end
end

function set.have(s, key)
  if s[key] == nil then
    return false
  else
    return true
  end
end

-- you need to define a function to sort values in ascending order.
function f(currenyVal, nextVal)
  return currenyVal < nextVal -- number
end

s = {}
set.insert(s, 1, f)
set.insert(s, 3, f)
set.insert(s, 9, f)
set.insert(s, 5, f)

print(set.have(s, 10)) -- 判断有无
print(set.have(s, 1))

for key, value in ipairs(s) do
  print(key, value)
end

table as linked lists

请参考：https://www.lua.org/pil/11.3.html

table as queues （队列，先进先出，不做介绍，请参考原文）

请参考：https://www.lua.org/pil/11.4.html

参考

http://blog.51cto.com/rangercyh/1032925

https://www.lua.org/pil/11.html