2161 根据给定数字划分数组（模拟、双指针）

1. 问题描述：

给你一个下标从 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 <= 10 ^ 5
-10 ^ 6 <= nums[i] <= 10 ^ 6
pivot 等于 nums 中的一个元素。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/partition-array-according-to-given-pivot/

2. 思路分析：

分析题目可以知道我们需要使得小于pivot的元素在pivot左边，大于pivot的元素在pivot右边，并且原来数组中元素的相对位置不变，一个比较容易想到方法声明两个记录数组a，b分别记录小于和大于pivot的元素，声明一个变量x来记录等于pivot元素的数目，最后将记录的结果填充到res数组中即可；除了这种方法之外我们还可以使用双指针来解决，声明两个指针l，r分别位于nums开始与结束的位置，为了避免处理等于pivot元素的问题我们可以一开始将答案数组中的所有元素填充为pivot，遍历nums将小于pivot的元素填充到答案数组对应指针l的位置上，大于pivot的元素填充到答案数组对应指针r的位置上，并且l右移，r左移，最终小于pivot的元素都填充到了res的左边，大于pivot的元素都填充到了res的右边，由于是从前往后遍历nums，所以还需要将大于pivot的所有元素再翻转一下相对位置才是正确的。

3. 代码如下：

python：

from typing import List


class Solution:
    def pivotArray(self, nums: List[int], pivot: int) -> List[int]:
        n = len(nums)
        res = [pivot] * n
        l, r = 0, n - 1
        for x in nums:
            if x > pivot:
                res[r] = x
                r -= 1
            elif x < pivot:
                res[l] = x
                l += 1
        x, y = r + 1, n - 1
        while x < y:
            res[x], res[y] = res[y], res[x]
            x += 1
            y -= 1
        return res

go：

package main

import "fmt"

func pivotArray(nums []int, pivot int) []int {
	n := len(nums)
	res := make([]int, n)
	for i := 0; i < n; i++ {
		res[i] = pivot
	}
	l, r := 0, n-1
	for i := 0; i < n; i++ {
		if nums[i] < pivot {
			res[l] = nums[i]
			l += 1
		} else if nums[i] > pivot {
			res[r] = nums[i]
			r -= 1
		}
	}
	// 翻转一下大于当前pivot元素的元素
	x, y := r+1, n-1
	for x < y {
		res[x], res[y] = res[y], res[x]
		x += 1
		y -= 1
	}
	return res
}

package main

import "fmt"

func pivotArray(nums []int, pivot int) []int {
	x := 0
	rec1, rec2 := []int{}, []int{}
	for i := 0; i < len(nums); i++ {
		if nums[i] < pivot {
			rec1 = append(rec1, nums[i])
		} else if nums[i] > pivot {
			rec2 = append(rec2, nums[i])
		} else {
			x += 1
		}
	}
	res := []int{}
	for i := 0; i < len(rec1); i++ {
		res = append(res, rec1[i])
	}
	for i := 0; i < x; i++ {
		res = append(res, pivot)
	}
	for i := 0; i < len(rec2); i++ {
		res = append(res, rec2[i])
	}
	return res
}