数据结构——移除元素

先献上题目：力扣https://leetcode.cn/problems/remove-element/

目录

1、时间复杂度O(n^2)，空间复杂度O(1) 

2、时间复杂度O(n)，空间复杂度O(n)

3、时间复杂度O(n)，空间复杂度O(1)

---

1、时间复杂度O(n^2)，空间复杂度O(1) 

题意很简单，就是移除一个数组中特定的元素，那么我们首先想到的办法可能是我遍历这个数组，当我遇到特定元素时候要想办法移除这个元素，我让后面的元素顶上来，就相当于是移除掉了这个元素。代码实现就是

int removeElement(int* nums, int numsSize, int val)
{
    for(int i=0; i < numsSize; i++)
    {
        if(nums[i] == val)
        {
            for(int j=i;j<numsSize-1;j++)
            {
                nums[j]=nums[j+1];
            }
            numsSize--;
            i--;
        }
    }
    return numsSize;
}

实际上这段代码的时间复杂度O(n^2)算是比较差的了，而且在写代码时候对数组边界的处理可谓是相当麻烦。所以这个办法就不是很好用。下面介绍一种用空间换时间的方法：

2、时间复杂度O(n)，空间复杂度O(n)

这种方法的思路是我创建一个和nums对应数组一样大的数组，然后遍历这个数组，如果遇到不等于val的元素，那我就把它拷贝进创建的数组里，最后再把创建的数组拷贝进nums对应数组里面，这样就大大降低了时间复杂度，我只需要遍历一遍，不需要再循环地去移动元素。但是因为创建了新的数组，所以空间复杂度就上升了。

int removeElement(int* nums, int numsSize, int val)
{
    int arr[numsSize];
    int* p = arr;
    for(int i = 0; i < numsSize; i++)
    {
        if(nums[i] != val)
        {
            *p = nums[i];
            p++;
        }
    }
    memmove(nums,arr,numsSize*sizeof(*nums));
    return p - arr;
}

3、时间复杂度O(n)，空间复杂度O(1)

基于我们刚才第二种方法，如果我们不创建新的数组，用两个指针，遇到等于val的元素其中一个不动，另一个继续走，继续走的指针通过判断元素，不等于val就放进走得慢的指针里面，从而实现自己改自己。

int removeElement(int* nums, int numsSize, int val)
{
    int* p1 = nums;
    int* p2 = nums;
    for(int i = 0; i < numsSize; i++)
    {
        if(*p1 != val)
        {
            *p2 = *p1;
            p2++;
        }
        p1++;
    }
    return p2 - nums;
}

这样就完美解决了移除元素的问题了。