关关的刷题日记15 – Leetcode 27. Remove Element 方法1、2、3

题目

Given an array and a value, remove all instances of that value in place and return the new length. Do not allocate extra space for another array, you must do this in place with constant memory. The order of elements can be changed. It doesn't matter what you leave beyond the new length. Example: Given input array nums = [3,2,2,3], val = 3 Your function should return length = 2, with the first two elements of nums being 2.

题目的意思是给定一个数组和一个值，让在不开辟额外空间的情况下删掉数组中该值，并且返回除去该目标值之后数组的新长度。数组中元素的顺序可以发生变化。

方法1：最容易想到的，就是把目标值都删掉，然后返回数组的长度。每次erase的代价是O(n)，算法时间复杂度O(n2)。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        for(int i=0; i<nums.size(); i++)
        {
            if(nums[i]==val)
            {
               nums.erase(nums.begin()+i);
                i--;
            }           
        }
        return nums.size();
    }
};

方法2：为了降低时间复杂度，可以先排个序，将目标值集中在一处进行批量删除，批量删除erase的时间复杂度是O(n)，算法的时间复杂度是O(nlogn)。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        sort(nums.begin(),nums.end());
        int flag=0, flag1,count=0;
        for(int i=0; i<nums.size(); i++)
        {
            if(nums[i]==val)
            {
                if(flag==0)
                {
                    flag1=i;
                    flag=1;
                }
                count++;
            }           
        }
        nums.erase(nums.begin()+flag1,nums.begin()+flag1+count);
        return nums.size();
    }
};

方法3：题目要求不能开辟新的空间，所以想到设置两个头尾指针进行元素互换，将所有目标值都换到数组的尾部。时间复杂度O(n)

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        if(nums.empty())
            return 0;
        int temp=0;
        for(int i=0, j=nums.size()-1; i<j;)
        {
            while(i<nums.size() && nums[i]!=val)
                i++;
            while(j>=0 && nums[j]==val)
                j--;
            if(i<j)
            {
               swap(nums[i],nums[j]);
            }
        }
        for(int i=0; i<nums.size(); i++)
        {
            if(nums[i]==val)
                return i;
        }
        return nums.size();
    }
};

vector中erase的用法如下： iterator erase( iterator _Where); iterator erase( iterator _First, iterator _Last); 如果是删除指定位置的元素时： 返回值是一个迭代器，指向删除元素下一个元素; 如果是删除某范围内的元素时：返回值也表示一个迭代器，指向最后一个删除元素的下一个元素;

人生易老，唯有陪伴最长情，加油！

以上就是关关关于这道题的总结经验，希望大家能够理解，有什么问题可以在我们的专知公众号平台上交流或者加我们的QQ专知-人工智能交流群 426491390，也可以加入专知——Leetcode刷题交流群（请先加微信小助手weixinhao: Rancho_Fang）。 同时请，关注我们的公众号，获取最新关于专知以及人工智能的资讯、技术、算法等内容。扫一扫下方关注我们的微信公众号。