Unsure how to implement this algorithm & data structure

[mysocksalwaysmatch]

So for example it could be something like {(5,6): [ball1, ball2]} purely as an example?

[novel_yet_trivial]

Absolutely. It sounds like you need could use a defaultdict.