Solution in python of leetcode 384 Shuffle an Array