LeetCode | Remove Element

解法一：空间复杂度O(n)

 

使用额外的空间可以使思路变得简单，要做的就是从A里面逐一拷贝元素到aux里，其中跳过elem就行了。缺点就是提升了空间复杂度，还有到最后需要把aux

 

的值在重新赋值给A，这样时间复杂度也多了一倍到达了theta(2n)，空间复杂度为O(n)

 

复制代码

int removeElement2(int A[], int n, int elem) {    int len = n;    int aux[n];    int k = 0;    for (int i = 0; i < n; i++) {        if (A[i] == elem) {            len--;        }        else {            aux[k++] = A[i];        }    }        for (int i = 0; i < len; i++) {        A[i] = aux[i];    }    return len;}

复制代码

 

 

解法二：认真观察删除操作的性质会发现，删除一个元素不过就是把他后面的元素向前移动一位，那删除两个不就是移动两位，三个就是三位。。。以此类推。

 

所以我们只用在遍历数组的时候跟踪删除的个数n，然后同时将元素向后挪动n位就可以了，就这么简单。

 

复制代码

int removeElement(vector<int>& nums, int val) {        //在遍历数组的时候跟踪删除的个数n，然后同时将元素向后挪动n位就可以了        int cnt=0;        int size=nums.size();        for(int i=0;i<size;i++){            if(nums[i]==val)                cnt++;            else if(cnt>0)                nums[i-cnt]=nums[i];        }        return size-cnt;    }