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

题目描述

给你一个下标从 0 开始的整数数组 nums 和一个整数 pivot 。请你将 nums 重新排列,使得以下条件均成立:

  • 所有小于 pivot 的元素都出现在所有大于 pivot 的元素 之前
  • 所有等于 pivot 的元素都出现在小于和大于 pivot 的元素 中间
  • 小于 pivot 的元素之间和大于 pivot 的元素之间的 相对顺序 不发生改变。
    • 更正式的,考虑每一对 pipjpi 是初始时位置 i 元素的新位置,pj 是初始时位置 j 元素的新位置。对于小于 pivot 的元素,如果 i < jnums[i] < pivotnums[j] < pivot 都成立,那么 pi < pj 也成立。类似的,对于大于 pivot 的元素,如果 i < jnums[i] > pivotnums[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相等。

以下是代码的详细解析:

  1. 定义三个空列表:leftrightbetween。这三个列表将分别存储小于pivot的元素、大于pivot的元素和等于pivot的元素。

  2. 遍历输入的列表nums中的每个元素。对于每个元素,根据其值与pivot的比较结果,将其添加到相应的列表中:

    • 如果元素的值小于pivot,则将其添加到left列表中。
    • 如果元素的值大于pivot,则将其添加到right列表中。
    • 如果元素的值等于pivot,则将其添加到between列表中。
  3. 最后,将这三个列表连接起来并返回结果。这个结果是一个新的列表,其中包含按照要求排序的元素。

python代码

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

运行结果

相关推荐
Pluchon1 分钟前
硅基计划4.0 算法 简单模拟实现位图&布隆过滤器
java·大数据·开发语言·数据结构·算法·哈希算法
AIFarmer2 分钟前
在EV3上运行Python语言——环境设置
python·ev3
独断万古他化7 分钟前
【算法通关】前缀和:和为 K、和被 K整除、连续数组、矩阵区域和全解
算法·前缀和·矩阵·哈希表
yunsr7 分钟前
python作业3
开发语言·python
历程里程碑9 分钟前
普通数组-----除了自身以外数组的乘积
大数据·javascript·python·算法·elasticsearch·搜索引擎·flask
曦月逸霜9 分钟前
Python快速入门——学习笔记(持续更新中~)
笔记·python·学习
静听山水11 分钟前
Redis核心数据结构-list
数据结构·redis·list
AI视觉网奇11 分钟前
blender 导入fbx 黑色骨骼
学习·算法·ue5·blender
喵手11 分钟前
Python爬虫实战:采集菜谱网站的“分类/列表页”(例如“家常菜”或“烘焙”频道)数据,构建高可用的美食菜谱数据采集流水线(附CSV导出)!
爬虫·python·爬虫实战·零基础python爬虫教学·采集菜谱网站数据·家常菜或烘焙频道·构建高可用食谱数据采集系统
喵手13 分钟前
Python爬虫实战:硬核解析 Google Chrome 官方更新日志(正则+文本清洗篇)(附 CSV 导出)!
爬虫·python·爬虫实战·零基础python爬虫教学·csv导出·监控谷歌版本发布历史·获取稳定版更新日志