题目描述
给你一个下标从 0 开始的整数数组 nums
和一个整数 pivot
。请你将 nums
重新排列,使得以下条件均成立:
- 所有小于
pivot
的元素都出现在所有大于pivot
的元素 之前 。 - 所有等于
pivot
的元素都出现在小于和大于pivot
的元素 中间 。 - 小于
pivot
的元素之间和大于pivot
的元素之间的 相对顺序 不发生改变。- 更正式的,考虑每一对
pi
,pj
,pi
是初始时位置i
元素的新位置,pj
是初始时位置j
元素的新位置。对于小于pivot
的元素,如果i < j
且nums[i] < pivot
和nums[j] < pivot
都成立,那么pi < pj
也成立。类似的,对于大于pivot
的元素,如果i < j
且nums[i] > pivot
和nums[j] > pivot
都成立,那么pi < pj
。
- 更正式的,考虑每一对
请你返回重新排列 nums
数组后的结果数组。
示例 1:
输入:nums = [9,12,5,10,14,3,10], pivot = 10
输出:[9,5,3,10,10,12,14]
解释:
元素 9 ,5 和 3 小于 pivot ,所以它们在数组的最左边。
元素 12 和 14 大于 pivot ,所以它们在数组的最右边。
小于 pivot 的元素的相对位置和大于 pivot 的元素的相对位置分别为 [9, 5, 3] 和 [12, 14] ,它们在结果数组中的相对顺序需要保留。
示例 2:
输入:nums = [-3,4,3,2], pivot = 2
输出:[-3,2,4,3]
解释:
元素 -3 小于 pivot ,所以在数组的最左边。
元素 4 和 3 大于 pivot ,所以它们在数组的最右边。
小于 pivot 的元素的相对位置和大于 pivot 的元素的相对位置分别为 [-3] 和 [4, 3] ,它们在结果数组中的相对顺序需要保留。
提示:
1 <= nums.length <= 105
-106 <= nums[i] <= 106
pivot
等于nums
中的一个元素。
解析
目标是将给定的整数列表nums
根据指定的pivot
值进行划分,使得左边的元素都小于pivot
,右边的元素都大于pivot
,而中间的元素则与pivot
相等。
以下是代码的详细解析:
-
定义三个空列表:
left
、right
和between
。这三个列表将分别存储小于pivot
的元素、大于pivot
的元素和等于pivot
的元素。 -
遍历输入的列表
nums
中的每个元素。对于每个元素,根据其值与pivot
的比较结果,将其添加到相应的列表中:- 如果元素的值小于
pivot
,则将其添加到left
列表中。 - 如果元素的值大于
pivot
,则将其添加到right
列表中。 - 如果元素的值等于
pivot
,则将其添加到between
列表中。
- 如果元素的值小于
-
最后,将这三个列表连接起来并返回结果。这个结果是一个新的列表,其中包含按照要求排序的元素。
python代码
python
class Solution:
def pivotArray(self, nums: List[int], pivot: int) -> List[int]:
left=[]
right=[]
between=[]
for x in nums:
if x<pivot:
left.append(x)
elif x>pivot:
right.append(x)
else:
between.append(x)
return left+between+right