48 lines
776 B
Lua
48 lines
776 B
Lua
io.input("test")
|
|
io.input("input")
|
|
|
|
calories = {0}
|
|
i = #calories
|
|
for line in io.lines() do
|
|
if line ~= "" then
|
|
calories[i] = calories[i] + tonumber(line)
|
|
else
|
|
table.insert(calories, 0)
|
|
i = #calories
|
|
end
|
|
end
|
|
|
|
max = 0
|
|
maxa, maxb, maxc = 0, 0, 0
|
|
for i, c in ipairs(calories) do
|
|
if c>max then
|
|
max = c
|
|
end
|
|
if c>maxa then
|
|
maxc = maxb
|
|
maxb = maxa
|
|
maxa = c
|
|
elseif c>maxb then
|
|
maxc = maxb
|
|
maxb = c
|
|
elseif c>maxc then
|
|
maxc = c
|
|
end
|
|
-- print(c)
|
|
end
|
|
|
|
part1 = maxa
|
|
part2 = maxa+maxb+maxc
|
|
assert(part1==71506)
|
|
assert(part2==209603)
|
|
print("part 1", part1)
|
|
print("part 2", part2)
|
|
|
|
-- using table.sort
|
|
table.sort(calories, function(a,b) return (a>b) end)
|
|
part1 = calories[1]
|
|
part2 = calories[1]+calories[2]+calories[3]
|
|
print("part 1", part1)
|
|
print("part 2", part2)
|
|
|