【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++;
}
}
}