数组 | 27. 移除元素

零、描述

题目链接：27. 移除元素

一、想法：

首先想用两个指针，左边指针从头到尾，右边从尾到头。
当左边指针指向arr[ i ] != target 时候，记录下索引 L，
右指针开始从尾到头，当arr[ j ]==target时候，记录下索引 R 。

int removeElement(vector<int>& nums, int val) {
    
    
        int N=nums.size();
        int l=0,r=N-1;     //存储下一次循环的地点
        if(N==0) return {
    
    };
  
        for(int i=l;i<r;i++){
    
    
            if(nums[i]==val){
    
    
                l=i+1;       //留作下次开始
                for(int j=r;j>=i;j--){
    
    
                    if(nums[j]!=val){
    
    
                        r=j-1;
                        nums[i]=nums[j];
                        break;
                    }
                }
            }
        }
        return l+1;
    }

当输入nums=[1] val=1 时候，正确结果是[ ] ,但是代码返回的结果是 [1]

二、参考

代码随想录27-移除元素

/**
* 相向双指针方法，基于元素顺序可以改变的题目描述改变了元素相对位置，确保了移动最少元素
* 时间复杂度：O(n)
* 空间复杂度：O(1)
*/
int removeElement(vector<int>& nums, int val) {
    
    
        int leftIndex = 0;
        int rightIndex = nums.size() - 1;
        
        while(leftIndex<=rightIndex){
    
    
            // 找左边等于val的元素
            while (leftIndex <= rightIndex && nums[leftIndex] != val){
    
    
                ++leftIndex;
            }
            //跳出while时候，leftIndex表示此时左边等于val的元素
            
            // 找右边不等于val的元素
            while (leftIndex <= rightIndex && nums[rightIndex] == val) {
    
    
                -- rightIndex;
            }
            //跳出while时候，leftIndex表示此时右边不等于val的元素
            
            // 将右边不等于val的元素覆盖左边等于val的元素
            if (leftIndex < rightIndex) {
    
    
                nums[leftIndex++] = nums[rightIndex--];
            }
        }
        return leftIndex;   // leftIndex一定指向了最终数组末尾的下一个元素
    }
};

举例

nums = [3,2,2,3], val = 3
leftIndex=0, rightIndex=3
nums[ leftIndex=0 ] == val, leftIndex=0
nums[ rightIndex=3 ] == val, rightIndex=2
nums[ rightIndex=2 ] ！= val 跳出while
到底if ,此时leftIndex=0, rightIndex=2 交换得到nums = [2,2,3,3]

总结

• 采用 while 替代 for
• while实现是也是 if 中符合条件停止左指针 的功能