力扣 第27题:移除元素(Java)

这道题是数组数据的删除操作,是一道比较简单的算法题。用双指针就可以实现空间复杂度为O(1)的操作,先看一下题目吧:

给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例 :
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2]

思路:题目告诉我们这道题要求在原数组中进行删除操作,仅能使用O(1)额外的数组空间。我的第一反应是建一个新数组,并导入原数组中除val以外的其他元素。但O(1)的额外空间让我不能这么操作,只能在原数组上进行元素移动或移除了。
双指针是满足O(1)空间复杂度的操作方式。其逻辑是创建两个指针,使指针A从数组头部开始遍历,如果A指针指向的元素值不为val,则将其值赋给B指针,使B指针同时移动;若A遍历到了值为var的元素,则不保留该元素,A指针继续走,B指针原地不动。这样A指针遍历完一遍后就可以将所有的val元素从数组中剔除掉,最坏情况下数组中没有val元素,B指针需要遍历完整个数组。代码如下:

class Solution {
    public int removeElement(int[] nums, int val) {
        if(nums == null)return 0;
        int len = nums.length;
        int j = 0;
        for(int i = 0;i<len;i++){
            if(nums[i]!=val){
                nums[j] = nums[i];
                j++;
            }
        }return j;
    }
}

这个方法已经满足题目需求了,空间复杂度为O(1),但最坏情况下需要遍历两遍数组,这一点我们可以从指针的位置上来优化一下。

优化的方法也比较简单,同样是两个指针,指针A从头开始遍历,指针B则从数组尾部开始遍历。当指针A遍历到val元素时,用指针B的元素将其替换掉,当两个指针重合时结束遍历。通过这种方法,在任何情况下我们都只需要遍历一遍数组。


class Solution {
    public int removeElement(int[] nums, int val) {
        int left = 0;
        int right = nums.length;
        while (left < right) {
            if (nums[left] == val) {
                nums[left] = nums[right - 1];
                right--;
            } else {
                left++;
            }
        }
        return left;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值