Implemented Thomas Algorithm in Lua

Author: Vexatos

CONTRIBUTORS.md

@@ -38,3 +38,4 @@
 - PaddyKe
 - nic-hartley
 - crafter312
+- Vexatos

contents/thomas_algorithm/code/lua/thomas.lua

@@ -0,0 +1,36 @@
+function thomas(a, b, c, d)
+
+  -- Create tables and set initial elements
+  local c_prime = {c[1] / b[1]}
+  local result = {d[1] / b[1]}
+
+  for i = 2, #a do
+    -- Scale factor is for c_prime and result
+    local scale = 1.0 / (b[i] - a[i] * c_prime[i - 1])
+    c_prime[i] = c[i] * scale
+    result[i] = (d[i] - a[i] * result[i - 1]) * scale
+  end
+
+  -- Back-substitution
+  for i = #a-1, 1, -1 do
+    result[i] = result[i] - (c_prime[i] * result [i + 1])
+  end
+
+  return result
+end
+
+local a = {0.0, 2.0, 3.0}
+local b = {1.0, 3.0, 6.0}
+local c = {4.0, 5.0, 0.0}
+local d = {7.0, 5.0, 3.0}
+
+print("The system")
+print(table.unpack(a))
+print(table.unpack(b))
+print(table.unpack(c))
+print(table.unpack(d))
+print("Has the solution:")
+
+local solution = thomas(a, b, c, d)
+
+print(table.unpack(solution))

contents/thomas_algorithm/thomas_algorithm.md

@@ -121,6 +121,8 @@ You will find this algorithm implemented [in this project](https://scratch.mit.e
 [import, lang:"nim"](code/nim/thomas_algorithm.nim)
 {% sample lang="cpp" %}
 [import, lang:"c_cpp"](code/c++/thomas.cpp)
+{% sample lang="lua" %}
+[import, lang:"lua"](code/lua/thomas.lua)
 {% endmethod %}
 
 <script>