class Solution {
public:
vector<int> pivotArray(vector<int>& nums, int pivot) {
}
};
2161. 根据给定数字划分数组
给你一个下标从 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] <= 106pivot 等于 nums 中的一个元素。原站题解
python3 解法, 执行用时: 164 ms, 内存消耗: 28.5 MB, 提交时间: 2022-11-12 13:08:44
class Solution:
def pivotArray(self, nums: List[int], pivot: int) -> List[int]:
n = len(nums)
ans = [pivot] * n
left, right = 0, n - 1
for i in range(n):
if nums[i] < pivot:
ans[left] = nums[i]
left += 1
elif nums[i] > pivot:
ans[right] = nums[i]
right -= 1
x, y = right + 1, n - 1
while x < y:
ans[x], ans[y] = ans[y], ans[x]
x += 1
y -= 1
return ans