目标
给你一个下标从 0 开始的整数数组 nums 和一个整数 pivot 。请你将 nums 重新排列,使得以下条件均成立:
- 所有小于 pivot 的元素都出现在所有大于 pivot 的元素 之前 。
- 所有等于 pivot 的元素都出现在小于和大于 pivot 的元素 中间 。
- 小于 pivot 的元素之间和大于 pivot 的元素之间的 相对顺序 不发生改变。
- 更正式的,考虑每一对 pi,pj ,pi 是初始时位置 i 元素的新位置,pj 是初始时位置 j 元素的新位置。如果 i < 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 <= 10^5
- -10^6 <= nums[i] <= 10^6
- pivot 等于 nums 中的一个元素。
思路
根据给定数字 pivot 划分数组,左边的都小于 pivot,右边的都大于 pivot,中间的等于 pivot,同一边元素的相对位置不能改变。
直接使用两个列表保存 小于 和 大于 pivot 的元素,计算等于 pivot 的元素个数,按顺序回填原数组即可。
代码
/**
* @date 2026-06-08 9:49
*/
public class PivotArray2161 {
public int[] pivotArray(int[] nums, int pivot) {
int n = nums.length;
Deque<Integer> l = new ArrayDeque<>();
Deque<Integer> r = new ArrayDeque<>();
for (int num : nums) {
if (num < pivot) {
l.offer(num);
} else if (num > pivot) {
r.offer(num);
}
}
int equal = n - l.size() - r.size();
int i = 0;
while (!l.isEmpty()) {
nums[i++] = l.poll();
}
while (equal > 0){
nums[i++] = pivot;
equal--;
}
while (!r.isEmpty()) {
nums[i++] = r.poll();
}
return nums;
}
}
性能









