题目:
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 :必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]
示例 2:
输入: nums = [0]
输出: [0]
思路及解法
使用双指针,左指针指向当前已经处理好的序列的尾部,右指针指向待处理序列的头部。
右指针不断向右移动,每次右指针指向非零数,则将左右指针对应的数交换,同时左指针右移。
注意到以下性质:
左指针左边均为非零数;
右指针左边直到左指针处均为零。
因此每次交换,都是将左指针的零与右指针的非零数交换,且非零数的相对顺序并未改变。
代码:
public void moveZeroes(int[] nums) {
int length = nums.length;
int left = 0;
int right = 0;
while (right < length) {
if (nums[right] != 0) {
swap(nums, left, right);
left++;
}
right ++;
}
}
public void swap(int[] nums, int left, int right) {
int temp = nums[left];
nums[left] = nums[right];
nums[right] = temp;
}
@Test
public void test2(){
//nums = [0,1,0,3,12]
int[] nums1 = {0,1,0,3,12};
moveZeroes(nums1);
for (int i : nums1) {
System.out.println(i);
}
// 1 3 12 0 0
System.out.println("--------------");
//nums = [1,5,0,3,2];
int[] nums2 = {1,5,0,3,2};
moveZeroes(nums2);
for (int i : nums2) {
System.out.println(i);
}
// 1 5 3 2 0
}
复杂度分析:
时间复杂度:O(n),其中 n 为序列长度。每个位置至多被遍历两次。
空间复杂度:O(1)。只需要常数的空间存放若干变量。