Implement Equilibrium Index problem.

Not too happy with how the `filter_map` part turned out. Suggestions to improve are welcome!