2021/day12.m4

   1 divert(-1)dnl -*- m4 -*-
   2 # Usage: m4 [-Dfile=day12.input] day12.m4
   3
   4 include(`common.m4')ifelse(common(12), `ok', `',
   5 `errprint(`Missing common initialization
   6 ')m4exit(1)')
   7
   8 # `n'ode name->number, `a'djacent neighbors, `s'mall
   9 define(`count', 0)
  10 define(`_node', `define(`n$1', $2)define(`s$2', eval(len(translit(alpha,
  11   `$1')) < 26))')
  12 define(`node', `ifdef(`n$1', `', `_$0(`$1', eval(1<<count))define(`count',
  13   incr(count))')')
  14 define(`adj', `ifelse(`$4', `start', `', `$2', `end', `', `define(`a$1',
  15   defn(`a$1')`,$3')')')
  16 define(`parse', `ifelse(`$1', `', `', `node(`$1')node(`$2')adj(n$1, `$1',
  17   n$2, `$2')adj(n$2, `$2', n$1, `$1')$0(shift(shift($@)))')')
  18 parse(translit(include(defn(`file')), -nl, `,,'))
  19 node(`dbl')
  20
  21 define(`_visit', `_foreach(`+cache(', `, $2, $3)', a$1)')
  22 define(`visit', `ifelse($1, 'nend`, 1, eval($2&$1), 0, `0_$0($1,
  23   eval($2|($1*s$1)), $3)', $3eval($2&'ndbl`), 20, `0_$0($1,
  24   eval($2|''ndbl``), $3)', 0)')
  25 define(`cache', `ifdef(`c$3_$1_$2', `', `define(`c$3_$1_$2', eval(visit(
  26   $@)))')c$3_$1_$2`'')
  27 define(`part1', cache(nstart, 0, 1))
  28 define(`part2', cache(nstart, 0, 2))
  29
  30 divert`'part1
  31 part2