advent-of-code/2021/nim/day19.nim

229 lines
4.6 KiB
Nim
Raw Normal View History

2021-12-20 16:32:35 -06:00
import ./common, std/[sequtils, algorithm, sugar, sets, strformat, strutils, tables, options, json, hashes]
type
Vec3 = array[3, int]
2021-12-21 11:12:58 -06:00
Transform = array[3, (0..2, int)]
Beacons = seq[Vec3]
Scanner = (Option[Vec3], Beacons)
2021-12-20 16:32:35 -06:00
Vec3Pair = array[2, Vec3]
2021-12-21 11:12:58 -06:00
const TRANSFORMS: seq[Transform] = collect:
let x: 0..2 = 0
let y: 0..2 = 1
let z: 0..2 = 2
for xs in [-1, 1]:
for ys in [-1, 1]:
for zs in [-1, 1]:
for t in [[x, y, z], [x, z, y], [y, x, z], [y, z, x], [z, x, y], [z, y, x]]:
[(t[0], xs), (t[1], ys), (t[2], zs)]
2021-12-21 09:29:41 -06:00
2021-12-21 11:12:58 -06:00
proc `*`(v: Vec3, t: Transform): Vec3 = [v[t[0][0]] * t[0][1], v[t[1][0]] * t[1][1], v[t[2][0]] * t[2][1]]
proc `+`(a: Vec3, b: Vec3): Vec3 = [a[0] + b[0], a[1] + b[1], a[2] + b[2]]
proc `-`(a: Vec3, b: Vec3): Vec3 = [a[0] - b[0], a[1] - b[1], a[2] - b[2]]
2021-12-20 16:32:35 -06:00
proc parse(s: string): Vec3 =
let l = s.split(",", 3).map parseInt
[l[0], l[1], l[2]]
2021-12-21 11:12:58 -06:00
proc parse(l: openArray[string]): seq[Scanner] =
2021-12-20 16:32:35 -06:00
var sb: seq[Vec3] = @[]
for line in l:
if line == "":
2021-12-21 11:12:58 -06:00
result.add (none(Vec3), sb)
2021-12-20 16:32:35 -06:00
continue
elif line[0] == '-' and line[1] == '-':
sb = @[]
else: sb.add line.parse
2021-12-21 11:12:58 -06:00
result.add (none(Vec3), sb)
2021-12-20 16:32:35 -06:00
2021-12-21 11:12:58 -06:00
proc maybeGetScannerTransform(s1: Scanner, s2: Scanner): Option[(Vec3, Transform)] =
let (knownPos, beacons1) = s1
let (_, beacons2) = s2
for t in TRANSFORMS:
var dists = newCountTable[Vec3]()
for b1 in beacons1:
for b2 in beacons2:
let dist = b1 - (b2*t)
dists.inc dist
if dists[dist] >= 12:
return some (dist, t)
none (Vec3, Transform)
2021-12-20 16:32:35 -06:00
2021-12-21 11:12:58 -06:00
var scanners: seq[Scanner] = @[]
proc p1(input: seq[Scanner]): uint64 =
scanners = input
scanners[0][0] = some [0, 0, 0]
var known = @[0]
var needTransform = toSeq(1..<scanners.len)
while needTransform.len > 0:
block checker:
for ix in 0..<needTransform.len:
let i = needTransform[ix]
for ij in 0..<known.len:
let j = known[ij]
let mt = maybeGetScannerTransform(scanners[j], scanners[i])
if mt.isSome:
echo &"GOT ONE: {i} at {mt}"
let (offset, transform) = mt.get
echo &"Adding {i} (was at index {ix})"
known.add i
needTransform.delete(ix)
scanners[i][0] = some offset
scanners[i][1] = scanners[i][1].mapIt((it * transform) + offset)
break checker
var beaconPositions = toHashSet[Vec3] []
for s in scanners:
for b in s[1]:
beaconPositions.incl b
beaconPositions.len.uint64
proc p2(input: seq[Scanner]): uint64 =
for s1 in scanners:
let (p1o, _) = s1
for s2 in scanners:
let (p2o, _) = s2
let p1 = p1o.get
let p2 = p2o.get
result = max(result, (abs(p1[0]-p2[0]) + abs(p1[1]-p2[1]) + abs(p1[2]-p2[2])).uint64)
2021-12-20 16:32:35 -06:00
const input = """
--- scanner 0 ---
404,-588,-901
528,-643,409
-838,591,734
390,-675,-793
-537,-823,-458
-485,-357,347
-345,-311,381
-661,-816,-575
-876,649,763
-618,-824,-621
553,345,-567
474,580,667
-447,-329,318
-584,868,-557
544,-627,-890
564,392,-477
455,729,728
-892,524,684
-689,845,-530
423,-701,434
7,-33,-71
630,319,-379
443,580,662
-789,900,-551
459,-707,401
--- scanner 1 ---
686,422,578
605,423,415
515,917,-361
-336,658,858
95,138,22
-476,619,847
-340,-569,-846
567,-361,727
-460,603,-452
669,-402,600
729,430,532
-500,-761,534
-322,571,750
-466,-666,-811
-429,-592,574
-355,545,-477
703,-491,-529
-328,-685,520
413,935,-424
-391,539,-444
586,-435,557
-364,-763,-893
807,-499,-711
755,-354,-619
553,889,-390
--- scanner 2 ---
649,640,665
682,-795,504
-784,533,-524
-644,584,-595
-588,-843,648
-30,6,44
-674,560,763
500,723,-460
609,671,-379
-555,-800,653
-675,-892,-343
697,-426,-610
578,704,681
493,664,-388
-671,-858,530
-667,343,800
571,-461,-707
-138,-166,112
-889,563,-600
646,-828,498
640,759,510
-630,509,768
-681,-892,-333
673,-379,-804
-742,-814,-386
577,-820,562
--- scanner 3 ---
-589,542,597
605,-692,669
-500,565,-823
-660,373,557
-458,-679,-417
-488,449,543
-626,468,-788
338,-750,-386
528,-832,-391
562,-778,733
-938,-730,414
543,643,-506
-524,371,-870
407,773,750
-104,29,83
378,-903,-323
-778,-728,485
426,699,580
-438,-605,-362
-469,-447,-387
509,732,623
647,635,-688
-868,-804,481
614,-800,639
595,780,-596
--- scanner 4 ---
727,592,562
-293,-554,779
441,611,-461
-714,465,-776
-743,427,-804
-660,-479,-426
832,-632,460
927,-485,-438
408,393,-506
466,436,-512
110,16,151
-258,-428,682
-393,719,612
-211,-452,876
808,-476,-593
-575,615,604
-485,667,467
-680,325,-822
-627,-443,-432
872,-547,-609
833,512,582
807,604,487
839,-516,451
891,-625,532
-652,-548,-490
30,-46,-14
""".strip().split('\n').parse
2021-12-21 11:12:58 -06:00
doDayX 19, (n: int) => n.loadInput.parse, p1, p2, (input, 79'u64, 3621'u64)