数据结构与算法之 leetcode 数组移动元素 零元素移动优化

数据结构与算法阿木 发布于 2025-07-12 6 次阅读


数组移动元素(零元素移动优化)——LeetCode算法挑战

在LeetCode这个编程挑战平台上,数组移动元素问题是一个经典且具有挑战性的题目。题目要求我们将数组中的非零元素移动到数组的开头,而将零元素移动到数组的末尾。这个问题的核心在于如何高效地移动元素,同时保持数组中其他元素的相对位置不变。

问题分析

假设我们有一个数组 `nums`,其中包含非零和零元素。我们的目标是重排这个数组,使得所有非零元素都位于数组的前面,而所有零元素都位于数组的后面。例如,对于数组 `[0, 1, 0, 3, 12]`,我们的目标是将其重排为 `[1, 3, 12, 0, 0]`。

常规解法

最直观的解法是使用两个指针,一个从数组的前端开始,一个从数组的后端开始。我们向前移动前端的指针,直到找到一个零元素;同时向后移动后端的指针,直到找到一个非零元素。然后,我们交换这两个指针所指向的元素。重复这个过程,直到两个指针相遇或者前端指针超过了后端指针。

python

def moveZeroes(nums):


left, right = 0, len(nums) - 1


while left < right:


while left < right and nums[left] != 0:


left += 1


while left < right and nums[right] == 0:


right -= 1


if left < right:


nums[left], nums[right] = nums[right], nums[left]


left += 1


right -= 1


return nums


这种方法的时间复杂度为O(n),空间复杂度为O(1),因为它在原地修改数组。

优化解法

上述解法虽然简单,但我们可以进一步优化它。一种常见的优化方法是使用双指针技巧,但我们可以尝试减少不必要的比较和交换操作。

1. 双指针法(两遍扫描)

我们可以先遍历一次数组,统计非零元素的数量,然后使用这个数量来填充数组的前半部分。这种方法只需要遍历数组两次,因此时间复杂度为O(n)。

python

def moveZeroesOptimized(nums):


count = 0


for i in range(len(nums)):


if nums[i] != 0:


nums[count] = nums[i]


count += 1


for i in range(count, len(nums)):


nums[i] = 0


return nums


2. 双指针法(一次扫描)

我们可以使用一个额外的变量来记录下一个非零元素应该放置的位置。这种方法只需要遍历数组一次,因此时间复杂度为O(n)。

python

def moveZeroesOptimizedOnce(nums):


next_non_zero = 0


for i in range(len(nums)):


if nums[i] != 0:


nums[next_non_zero] = nums[i]


next_non_zero += 1


for i in range(next_non_zero, len(nums)):


nums[i] = 0


return nums


3. 快速排序法

我们可以将数组分为两部分:非零元素和零元素。然后,我们可以使用快速排序的分区方法来解决这个问题。这种方法的时间复杂度通常为O(nlogn),但平均情况下可能优于其他方法。

python

def moveZeroesQuickSort(nums):


def partition(left, right):


pivot = nums[right]


i = left


for j in range(left, right):


if nums[j] != 0:


nums[i], nums[j] = nums[j], nums[i]


i += 1


nums[i], nums[right] = nums[right], nums[i]


return i

def quickSort(left, right):


if left < right:


pivot_index = partition(left, right)


quickSort(left, pivot_index - 1)


quickSort(pivot_index + 1, right)

quickSort(0, len(nums) - 1)


return nums


总结

在LeetCode的数组移动元素问题中,我们可以使用多种方法来优化零元素的移动。从简单的双指针法到更高效的快速排序法,每种方法都有其优缺点。选择哪种方法取决于具体的应用场景和性能要求。通过理解这些不同的方法,我们可以更好地掌握数组操作和算法优化的技巧。