Baseball Game - Leetcode 682 - Python