Leetcode 27. Remove Element

27. Remove Element

• Total Accepted: 129784
• Total Submissions: 372225
• Difficulty: Easy

 

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.

 

思路：遇到val等值的数就删除。

代码：

这个适合于不改变原来数的顺序和位置的情况：

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

 

实际上，本题可以改变原来数的顺序，所以下面的写法也可以：

写法一：begin指针只记录可以留下的数。

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

 

写法二：将当前最后的元素替换掉确认已经重复的元素，再对置换后的数进行判断。

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

 

写法三：用count记录所有的不符合要求的数。

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