LeetCode 2161. 根据给定数字划分数组

文章目录

- 1. 题目
- 2. 解题

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. 解题

遍历一次，将3种数字取出来存在3个数组里
依次将3个数组的数字填回去

class Solution {
public:vector<int> pivotArray(vector<int>& nums, int pivot) {vector<int> a,b,c,ans;for(auto n : nums){if(n < pivot) a.push_back(n);else if(n == pivot) b.push_back(n);else c.push_back(n);}for(auto n : a) ans.push_back(n);for(auto n : b) ans.push_back(n);for(auto n : c) ans.push_back(n);return ans;}
};