Merge pull request #2422 from tylerw/tw/challenge-080-lua

Ch80: add Lua solutions

1 files changed, 25 insertions, 0 deletions

diff --git a/challenge-080/tyler-wardhaugh/lua/ch-2.lua b/challenge-080/tyler-wardhaugh/lua/ch-2.lua
new file mode 100755
index 0000000000..54d355d037
--- /dev/null
+++ b/challenge-080/tyler-wardhaugh/lua/ch-2.lua
@@ -0,0 +1,25 @@
+#!/usr/bin/env lua
+
+local t2 = {}
+
+function t2.count_candies(coll)
+  local count = #coll
+  local maybe_inc = function(i, j)
+    if (coll[i] > coll[j]) then count = count + 1 end
+  end
+
+  -- sweep left-to-right
+  for i=1,#coll-1 do maybe_inc(i, i+1) end
+  -- sweep right-to-left
+  for i=#coll,2,-1 do maybe_inc(i, i-1) end
+
+  return count
+end
+
+function t2.run(arg)
+  local N = {}
+  for _, v in ipairs(arg) do table.insert(N, tonumber(v)) end
+  print(t2.count_candies(N))
+end
+
+return t2