From 3d2be7b010a125678fd91406f01778c535420521 Mon Sep 17 00:00:00 2001
From: chirvasitua <chirvasitua@gmail.com>
Date: Wed, 14 Jul 2021 07:29:47 -0400
Subject: 1st commit on 079_lua

---
 challenge-079/stuart-little/lua/ch-2.lua | 35 ++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100755 challenge-079/stuart-little/lua/ch-2.lua

(limited to 'challenge-079/stuart-little/lua/ch-2.lua')

diff --git a/challenge-079/stuart-little/lua/ch-2.lua b/challenge-079/stuart-little/lua/ch-2.lua
new file mode 100755
index 0000000000..23247611bd
--- /dev/null
+++ b/challenge-079/stuart-little/lua/ch-2.lua
@@ -0,0 +1,35 @@
+#!/usr/bin/env lua
+
+-- run <script> <space-separated numbers>
+
+function histo(t)
+   local mx = math.max(table.unpack(t))
+   local h={}
+   for i=0,mx-1 do
+      local row={}
+      for k,v in ipairs(t) do
+	 table.insert(row,v < mx-i and " " or "#")
+      end
+      table.insert(h,row)
+   end
+   return h
+end
+
+function rain(t)
+   local q=0
+   for k,v in ipairs(t) do
+      if k > 1 and k < #t then
+	 local l = math.max(table.unpack(t,1,k-1))
+	 local r = math.max(table.unpack(t,k+1))
+	 local m = math.min(l,r)
+	 q=q+math.max(0,m-v)
+      end
+   end
+   return q
+end
+
+local nums={}
+for _,nr in ipairs(arg) do table.insert(nums,tonumber(nr)) end
+for _,row in ipairs(histo(nums)) do print(table.unpack(row)) end
+print(("-"):rep(20))
+print(rain(nums))
-- 
cgit