leetcode shuffle the array python