advent-of-code/2020/src/day3.nim

import streams

proc sled(s: Stream, velx: int, vely: int): int =
  var xpos, ypos: int
  for line in s.lines():
    inc ypos
    if (ypos - 1) mod vely > 0: continue
    if line[xpos mod line.len()] == '#': inc result
    xpos += velx

proc part1*(s: Stream): int = sled(s, 3, 1)

proc part2*(s: Stream): int =
  result = part1(s)
  for vels in [(1, 1), (5, 1), (7, 1), (1, 2)]:
    setPosition(s, 0)
    let (velx, vely) = vels
    result *= sled(s, velx, vely)