【LeetCode刷题】移动零

给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。

请注意 ,必须在不复制数组的情况下原地对数组进行操作。

示例 1:

复制代码
输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]

示例 2:

复制代码
输入: nums = [0]
输出: [0]

提示:

  • 1 <= nums.length <= 104
  • -231 <= nums[i] <= 231 - 1

方法思路

  1. 定义指针 k :用于记录非零元素应该放置的位置,初始为 0
  2. 遍历数组 :将所有非零元素按原相对顺序移动到 k 指向的位置,每放置一个非零元素,k 自增 1
  3. 补零操作 :遍历结束后,将 k 到数组末尾的所有位置赋值为 0,完成 "将 0 移到末尾" 的操作。

Python代码:

python 复制代码
from typing import List


class Solution:
    def moveZeroes(self, nums: List[int]) -> None:
        k = 0  # 记录非零元素的放置位置
        for i in range(len(nums)):
            if nums[i] != 0:
                nums[k] = nums[i]
                k += 1
        # 将剩余位置填充为0
        for i in range(k, len(nums)):
            nums[i] = 0


# 主程序用于测试
if __name__ == "__main__":
    # 测试示例1
    nums1 = [0, 1, 0, 3, 12]
    solution = Solution()
    solution.moveZeroes(nums1)
    print("示例1结果:", nums1)

    # 测试示例2
    nums2 = [0]
    solution.moveZeroes(nums2)
    print("示例2结果:", nums2)

LeetCode提交代码:

python 复制代码
class Solution:
    def moveZeroes(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        from typing import List
        k = 0  # 记录非零元素的放置位置
        for i in range(len(nums)):
            if nums[i] != 0:
                nums[k] = nums[i]
                k += 1
        # 将剩余位置填充为0
        for i in range(k, len(nums)):
            nums[i] = 0
        

程序运行截图展示:

总结

本文介绍了一种原地移动数组中所有零元素到末尾的算法。通过使用双指针法,定义指针k记录非零元素位置,首先遍历数组将所有非零元素前移,保持原有顺序;然后在剩余位置补零。该算法时间复杂度为O(n),空间复杂度O(1),满足题目要求。Python实现代码展示了示例测试结果,验证了算法的正确性。该方法高效地解决了在不复制数组情况下移动零元素的问题,适用于LeetCode等相关编程题目。

相关推荐
感谢地心引力1 分钟前
【AI】加入AI绘图的视频封面快速编辑器
人工智能·python·ai·ffmpeg·音视频·pyqt·gemini
Jerryhut3 分钟前
sklearn函数总结十二 —— 聚类分析算法K-Means
算法·kmeans·sklearn
我一定会有钱9 分钟前
pytest测试框架基础
python·单元测试·自动化·pytest
cr_每天进步一点点11 分钟前
【无标题】
人工智能·python·机器学习
我的xiaodoujiao15 分钟前
使用 Python 语言 从 0 到 1 搭建完整 Web UI自动化测试学习系列 35--二次封装MySQL数据库连接操作
python·学习·测试工具·pytest
DechinPhy16 分钟前
使用Python免费合并PDF文件
开发语言·数据库·python·mysql·pdf
驾数者18 分钟前
Flink SQL自定义函数开发:标量、聚合、表值函数实现
python·sql·flink
qq_2526144119 分钟前
python爬虫爬取视频
开发语言·爬虫·python
Swift社区23 分钟前
LeetCode 453 - 最小操作次数使数组元素相等
算法·leetcode·职场和发展
PNP Robotics23 分钟前
聚焦具身智能,PNP机器人展出力反馈遥操作,VR动作捕捉等方案,获得中国科研贡献奖
大数据·人工智能·python·学习·机器人