LeetCode 283:原地移动零的优雅解法

LeetCode 热题 283. 移动零:原地操作的优雅解法

一、题目理解:需求很明确

题目要求:

  • 把数组里的所有0移到末尾
  • 保持非零元素的相对顺序
  • 必须原地操作(不能复制新数组)

比如输入[0,1,0,3,12],输出得是[1,3,12,0,0];如果输入本身就是[0],直接返回[0]就行。

二、我的解题思路:两步走策略

一开始我想的是「双指针交换」,但后来发现更简洁的是 **"先填非零,再补零"** 的思路:

解法核心逻辑

  1. 第一步:把非零元素 "往前搬" 用一个变量stack_size记录 "非零元素应该放的位置",遍历数组时,遇到非零元素就放到nums[stack_size],然后stack_size自增。(相当于把所有非零元素按顺序 "堆" 在数组前半部分)

  2. 第二步:给剩下的位置补 0 遍历结束后,从stack_size的位置开始,把数组剩下的部分全填成 0。

代码实现(C++)

cpp

复制代码
class Solution {
public:
    void moveZeroes(vector<int>& nums) {
        int stack_size = 0;
        // 第一步:填充非零元素
        for (int x : nums) {
            if (x) {
                nums[stack_size++] = x;
            }
        }
        // 第二步:补0
        fill(nums.begin() + stack_size, nums.end(), 0);
    }
};

三、为什么这个解法高效?

  • 时间复杂度 O (n):只遍历数组两次(一次填非零,一次补零),n 是数组长度。
  • 空间复杂度 O (1):完全原地操作,没额外开数组。
  • 操作次数少:比起 "遇到 0 就交换" 的写法,减少了不必要的交换操作(比如连续多个 0 的情况)。

四、拓展:双指针交换法(另一种思路)

其实也可以用双指针 实现(比如left指针指向 "下一个非零元素该放的位置"):

cpp

复制代码
void moveZeroes(vector<int>& nums) {
    int left = 0;
    for (int i = 0; i < nums.size(); i++) {
        if (nums[i] != 0) {
            swap(nums[left], nums[i]);
            left++;
        }
    }
}

这个写法更 "紧凑",但本质和 "填非零 + 补零" 是一个逻辑 ------ 只是把 "赋值" 换成了 "交换"~

相关推荐
牛奔2 分钟前
Go语言中结构体转Map优雅实现
开发语言·后端·macos·golang·xcode
毕设源码-邱学长2 分钟前
【开题答辩全过程】以 台球俱乐部管理系统为例,包含答辩的问题和答案
java·eclipse
wujj_whut5 分钟前
【Conda实战】从0到1:虚拟环境创建、多Python版本管理与环境切换全指南
开发语言·python·conda
蜗牛^^O^7 分钟前
java中的JUC
java·开发语言
张心独酌8 分钟前
Rust新手练习案例库- rust-learning-example
开发语言·后端·rust
进阶小白猿20 分钟前
Java技术八股学习Day17
java·jvm·学习
小O的算法实验室25 分钟前
2024年ESWA SCI1区TOP,基于自适应模糊惩罚的多约束无人机路径规划状态转移算法,深度解析+性能实测
算法·论文复现·智能算法·智能算法改进
带刺的坐椅25 分钟前
从 Chat 到 Agent:Solon AI 带你进入“行动派”大模型时代
java·ai·agent·solon·mcp·java25
扶苏-su27 分钟前
Java--转换流-InputStreamReader 和 OutputStreamWriter
java·开发语言
我是小疯子6631 分钟前
深度学习实现智能文本摘要
java