1470. Shuffle the Array 链接到标题 重新排列数组,使用 zip 对两个切分后的数组进行聚合
class Solution: def shuffle(self, nums: List[int], n: int) -> List[int]: res = [] for i, j in zip(nums[0:n],nums[n:]): res+=[i,j] return res 1471. The k Strongest Values in an Array 链接到标题 找到与“中位数”数值相差最大的 k 个数字,先将数组排序,找到“中位数”,分别从开头和结尾两边进行比较,找到最大的 k个数。
class Solution: def getStrongest(self, arr: List[int], k: int) -> List[int]: arr.sort() i, j = 0, len(arr) - 1 median = arr[(len(arr) - 1) // 2] while len(arr) + i - j <= k: if median - arr[i] > arr[j] - median: i = i + 1 else: j = j - 1 return arr[:i] + arr[j + 1:] 1472.