LeetCode初级算法之数组：283 移动零

给定一个数组 nums，编写一个函数将所有 0 移动到数组的末尾，同时保持非零元素的相对顺序。

示例:

输入: [0,1,0,3,12] 

输出: [1,3,12,0,0]

说明:

1. 必须在原数组上操作，不能拷贝额外的数组。

2. 尽量减少操作次数。

题目信息

输入：数组

输出：数组（将原数组的0都移动到后面）

额外：空间O(1)，时间尽量减少

思考

对于这样一道题一开始看到要原地修改首先就想到了双指针遍历一次把不是零的往前设置。后面的再全部设置0，这样得到的算法也是满足空间复杂度O(1)，时间复杂度O(n).

 

public void moveZeroes(int[] nums) {
    int j = 0;
    for(int i=0;i<nums.length;i++){
        if(nums[i] != 0){
            nums[j]=nums[i];
            j++;
        }
    }
    while(j<nums.length){
        nums[j] = 0;
        j++;
    }
}

 

两次循环执行时间0ms，这样就已经算是完美解决了。但确实还可以只花一次循环就能解决，我们在遍历过程当中是保证其他顺序不变然后零到后面。上面的想法不就是扫描非0然后扫描一个就往前面设置然后指针移动一。那么同样我们去确定零的位置然后交换始终保证数字在最左零的左边即可

 

 

public void moveZeroes(int[] nums) {
    //两个指针i和j
    int j = 0;
    for(int i=0;i<nums.length;i++) {
        //当前元素!=0，就把其交换到左边，等于0的交换到右边
        if(nums[i]!=0) {
            int tmp = nums[i];
            nums[i] = nums[j];
            nums[j++] = tmp;
        }
    }
}

 

总结

总体来讲两种方式都是空间复杂度为O（1）的原地算法，时间复杂度也都为O（n）后者略优一点

 

本文分享自微信公众号 - IT那个小笔记（qq1839646816）。
如有侵权，请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”，欢迎正在阅读的你也加入，一起分享。