【算法】原地移除所有数值等于 val 的元素

最新推荐文章于 2023-12-14 13:41:42 发布

bbbbjt

最新推荐文章于 2023-12-14 13:41:42 发布

阅读量267

点赞数

分类专栏：算法笔记文章标签：算法指针 leetcode java 数据结构

本文链接：https://blog.csdn.net/bbbbjt/article/details/115891553

版权

算法笔记专栏收录该内容

1 篇文章 0 订阅

订阅专栏

力扣题库27. 移除元素

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

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

这里只需要用不等于val的元素替换掉已经遍历过的数组元素，返回新长度即可。

类似的26题删除有序数组中的重复项
给你一个有序数组 nums ，请你原地删除重复出现的元素，使每个元素只出现一次，返回删除后数组的新长度。
不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

class Solution {
    public int removeDuplicates(int[] nums) {
        int n = nums.length;
        if (n == 0) {
            return 0;
        }
        int fast = 1, slow = 1;
        while (fast < n) {
            if (nums[fast] != nums[fast - 1]) {
                nums[slow] = nums[fast];
                ++slow;
            }
            ++fast;
        }
        return slow;
    }
}

解法：由于数组是有序的，则重复的数组元素一定是连续的，可以利用双指针删除重复元素。定义两个指针 \textit{fast}fast 和 \textit{slow}slow 分别为快指针和慢指针，快指针表示遍历数组到达的下标位置，慢指针表示下一个不同元素要填入的下标位置，初始时两个指针都指向下标 1。