Bootstrap

【Leetcode-移动零】利用将非零元素移动至数组前解决移动零问题(剪枝优化)

题目描述

给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。

示例1

输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]

示例2

输入: nums = [0]
输出: [0]

解题思路 (时间复杂度O(n))

#include <stdlib.h>

void moveZeroes(int* nums, int numsSize) {
    int insertPos = 0;  // 使用insertPos变量记录非零元素应该插入的位置。

    for (int i = 0; i < numsSize; i++) { // 遍历数组,将非零元素交换到insertPos位置,并将insertPos递增。
        if (nums[i] != 0) {

            if (i != insertPos) { // 使用if (i != insertPos)条件判断,避免不必要的交换操作。
                int temp = nums[i];
                nums[i] = nums[insertPos];
                nums[insertPos] = temp;
            }
            insertPos++;
        }
    }
}

执行结果

在这里插入图片描述

;