当前位置：首页 > article >正文

蓝桥杯day01——根据给定数字划分数组

article 2024/11/19 3:29:34

题目描述

给你一个下标从 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代码

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