LeetCode删除数组元素系列问题总结

本文实现从数组中删除元素，提供时间复杂度为O(n)，空间复杂度为O(1)的Java版本的基本模板，在模板上修改，可以解决LeetCode中数组删除元素相关的问题，如：27. Remove Element（删除元素），26. Remove Duplicates from Sorted Array（删除排序数组中的重复项），80. Remove Duplicates from Sorted Array II（删除排序数组中的重复项II），283. Move Zeroes（移动零元素）。

数组中删除元素这个操作应该不会陌生，但是回答怎么删除的时候确实还是有些问题需要考虑清楚的，比如：

①如何定义删除？从数组中直接去除？还是放在数组的末尾位置？

②剩余元素的排列是否保证原有的相对顺序？

③是否有空间复杂度的要求？

今天在做27. Remove Element（删除元素）这个题的时候，我发现完全可以套用我之前在做LeetCode283. Move Zeroes时的方法，方法在之前的博客里面也有较为详细的介绍。下面直接上代码：

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

其中这段代码是完全可以复用的：

public void move(int[] nums) {
        int index = 0;
        for (int i = 0; i < nums.length; i++){
            if (){
                nums[index++] = nums[i];
            }
        }
}

27. Remove Element

第27题在nums数组中删除val的时候，只需要改变if的判断条件，如果不等于val的时候，将他添加进入数组，否则就跳过了，就是删除了。代码如下：

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

26. Remove Duplicates from Sorted Array

这个题需要删除数组中重复的元素，由于已经是有序的数组，所以重复的元素处在数组中相邻的位置。要删除重复的元素，只需要更改判断条件，相邻元素不等的情况下，讲元素添加进来，即可完成删除操作。代码如下：

class Solution {
    public int removeDuplicates(int[] nums) {
        int index = 1;
        for(int i=1;i<nums.length;i++){
            if(nums[i]!=nums[i-1]){
                nums[index++] = nums[i];
            }
        }
        return index;
    }
}

80. Remove Duplicates from Sorted Array II

这个题给定一个排序数组，删除重复出现的元素，使得每个元素最多出现两次，返回移除后数组的新长度。

这个题需要在26. Remove Duplicates from Sorted Array的基础上加上计数的功能，需要引入一个变量来记录元素出现的次数。如果数组中后一个元素等于前一个元素，并且这种情况（元素相等的情况）是第一次出现的时候，把这两个元素添加进去最终的数组，否则跳过去完成删除。代码如下：

class Solution {
    public int removeDuplicates(int[] nums) {
        int n = nums.length;
        if (n <= 2) return n;
        int repeat = 0, index = 1;
        for (int i = 1; i < n; ++i) {
            if (nums[i] == nums[i-1] && repeat < 1) {
                nums[index++] = nums[i];
                repeat++;
            }
            else if (nums[i] > nums[i-1]) {
                nums[index++] = nums[i];
                repeat = 0;
            }
        }
        return index;
    }
}