最近接觸了 Redis 腳本有使用到 Lua，簡單紀錄下基礎語法。

常用資料型態

運算符號

操作符號

變數

可一次宣告多個變數，沒指派值的會是 nil

local a = 1
local b, c, d = 2, 3

print(a)
print(b)
print(c)
print(d)
-- Output:
-- 1
-- 2
-- 3
-- nil

if else

if(boolean) then
   -- do...
else if(boolean) then
   -- do...
else 
   -- do...
end

Loop 系列

• for loop
  i 從 0 開始迴圈每次遞增 1 直到 i <= 3

for i = 0, 3, 1 do
  print(i)
end
-- Output:
-- 0
-- 1
-- 2
-- 3

• repeat until
  從 repeat 內開始跑迴圈直到 until 條件式成立就跳出

local i = 1
local count = 3
repeat
  print(i)
  i = i + 1
until i == count
-- Output:
-- 1
-- 2

• while do
  從 while 內開始跑迴圈直到條件式不成立就跳出

local i = 1
local count = 3
while i <= count do
  print(i)
  i = i + 1
end
-- Output:
-- 1
-- 2
-- 3

• ipairs
  ipairs 會按照陣列索引順序開始跑，直到 value 出現 nil 的情況

local a = { [1] =  "one", [2] = "two", [4] = "four" }
for index, value in ipairs(a) do
    print(index .. ":" .. value)
end
-- Output:
-- 1:one
-- 2:two

• pairs
  pairs 會按順序跑完全部陣列

local a = {
  [1] =  "one",
  [2] = "two",
  [4] = "four" ,
  dog = "狗",
  ["貓"] = "cat"
  }
for index, value in pairs(a) do
    print(index .. ":" .. value)
end
-- Output:
-- 1:one
-- 2:two
-- 4:four
-- dog:狗
-- 貓:cat

Function 系列

• 具有回傳值
  使用 return 回傳

function greet(name)
   local result = ""
   if (name == 'Max') 
   then
      result = 'Hello ' .. name
   else
      result = 'Who are you'
   end

   return result;
end

print(greet('Max'))
-- Output:
-- Hello Max

• 具有多回傳值
  使用 return 回傳，且用逗號分隔回傳值

function calculate(a, b)
  return a + b, a - b
end

local a, b = calculate(5, 23)
print('a + b = ' .. a)
print('a - b = ' .. b)
-- Output:
-- a + b = 28
-- a - b = -18

• 不具回傳值
  相當於 void

function greet(name)
   if (name == 'Max') 
   then
      print('Hello ' .. name)
   else
      print('Who are you')
   end
end

greet('someone')
-- Output:
-- Who are you

• 傳入任意參數
  傳入參數以逗號分隔

function add(...)
  local total = 0
  for index, value in ipairs{...} do
    total = total + value
  end
  
  return total
end

print(add(1,2,3))
-- Output:
-- 6

小試身手

題目：LeetCode 1. Two Sum

function TwoSum(nums, target)
  local table = { }
  for index, value in ipairs(nums) do
    if(table[target - value])
    then
      return table[target - value], index
    end
    table[value] = index
  end
end

local nums = {3,2,4} -- Sample
local target = 7 -- Sample
local one,two = TwoSum(nums, target)

print('[' .. one .. ',' .. two .. ']')

-- Input: nums = [3,2,4], target = 7

-- Output: [1,3]    
-- ※ Lua 的陣列索引從 1 起跳，所以原本答案是 [0,2] 會變為 [1,3]

參考

Lua 教程
撰寫和使用函式 (Function)
Lua | pairs與ipairs的差異
Tag: Lua | Level Up

• ← 上篇
• 下篇 →