Leetcode27.移除元素

问题

样例一：给定 nums = [3,2,2,3], val = 3,

函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。

你不需要考虑数组中超出新长度后面的元素。

样例二：给定 nums = [0,1,2,2,3,0,4,2], val = 2,

函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。

注意这五个元素可为任意顺序。

你不需要考虑数组中超出新长度后面的元素。

代码

双指针

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

         return j;
    }
};

 

---

 

更新一下

分析：这道题和 Leetcode26. 删除有序数组中的重复项  思路一致

采用双指针，k 代表结果数组指针，i表示遍历指针，遍历 i 找到符合条件的元素加入到k的位置，最后k会指向下一个元素

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
       int k = 0;
       for(int i = 0;i < nums.size();i++){
           if(i ==0 ||  nums[i] != nums[i-1]){
               nums[k++] = nums[i];
           }
       }
       return k;
    }
    
};