Wiggle Sort - Leetcode 280 - Python