题目如下:
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:
必须在原数组上操作,不能拷贝额外的数组。
尽量减少操作次数。
自己的算法及代码如下:
- 首先,选出数组中不为零的元素,并用 count 记录其个数,同时也作为新数组的下标;
- 然后,将不为0的元素向前移动,nums[count] = nums[i];所以得到的数组前 count 个元素都是不为零的,且保持了原数组中的相对位置;
- 最后,将后面剩余的元素赋值为0即可。
public class Solution {
public static void moveZeroes(int[] nums) {
int count = 0;
for (int i = 0; i < nums.length; i++){
if (nums[i] != 0){
nums[count] = nums[i];
count++;
}
}
for (int i = count; i < nums.length; i++){
nums[i] = 0;
}
// for (int i = 0; i < nums.length; i++){
// System.out.println(nums[i]);
// }
}
public static void main(String[] args) {
int[] nums = new int[]{
0, 1, 0, 3, 12};
moveZeroes(nums);
}
}