aboutsummaryrefslogtreecommitdiff
path: root/challenge-138/abigail/lua/ch-1.lua
blob: deae0602cfd519fe221029d8b2a90c0fe6240699 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#!/opt/local/bin/lua

--
-- See ../README.md
--

--
-- Run as: lua ch-1.lua < input-file
--

local SUNDAY    = 0
local MONDAY    = 1
local TUESDAY   = 2
local WEDNESDAY = 3
local THURSDAY  = 4
local FRIDAY    = 5
local SATURDAY  = 6

local lookup    = {
    {261, 260, 260, 261, 261, 261, 261},
    {262, 261, 260, 261, 262, 262, 262},
}

local anchors   = {TUESDAY, SUNDAY, FRIDAY, WEDNESDAY}

function doomsday (year)
    local anchor = anchors [1 + math . floor (year / 100) % 4]
    local y      = year % 100
    return ((math . floor (y / 12) + (y % 12) + math . floor ((y % 12) / 4)) +
             anchor) % 7
end

function is_leap (year)
    if (year % 400 == 0) or ((year % 4 == 0) and (year % 100 ~= 0)) then
        return 1
    end
    return 0
end


for year in io . lines () do
    year = tonumber (year)
    print (lookup [1 + is_leap (year)] [1 + doomsday (year)])
end