diff options
Diffstat (limited to '18/b.lua')
-rw-r--r-- | 18/b.lua | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/18/b.lua b/18/b.lua new file mode 100644 index 0000000..ab1c769 --- /dev/null +++ b/18/b.lua @@ -0,0 +1,62 @@ +input = io.lines('input.txt') + +function is_op(token) + return token == '+' or token == '*' +end + +function f_op(op) + if op == '+' then + return function (a, b) return a + b end + else + return function (a, b) return a * b end + end +end + +function is_digit(token) + return token >= '1' and token <= '9' +end + +function do_math(expression) + stack = {} + for i = 1, #expression do + token = expression:sub(i, i) + if is_op(token) then + table.insert(stack, token) + elseif is_digit(token) then + if stack[#stack] == '+' then + op = table.remove(stack) + n = table.remove(stack) + table.insert(stack, f_op(op)(n, tonumber(token))) + else + table.insert(stack, tonumber(token)) + end + elseif token == '(' then + table.insert(stack, token) + elseif token == ')' then + res = table.remove(stack) + last_token = table.remove(stack) + if last_token == '*' then + while not (last_token == '(') do + res = res * table.remove(stack) + last_token = table.remove(stack) + end + end + + if stack[#stack] == '+' then + table.remove(stack) + n = table.remove(stack) + table.insert(stack, res + n) + else + table.insert(stack, res) + end + end + end + return stack[1] +end + +sum = 0 +for line in input do + sum = sum + do_math('(' .. line .. ')') +end + +print(sum) |