It would have been simpler with a dictionary; I just expected part 2 to be harder than it was.

That was a fun one! I was worried that part 2 was going to end up needing something more performant. Glad I went for a simple solution in part 1 :)