数组类算法:移除元素

题目描述:

给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。

不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

题目解析:

使用双指针法,从两头开始遍历,遇到目标数字,进行left++或者right++等操作。

int RemoveElement(int *arr,int len,int elem)
{
    int left = 0;
    int right = len - 1;
    int remove_count = 0;

    while(left < right)
    {
        if(arr[left] == elem)
        {
            if(arr[right] == elem)
            {
                right --;
                remove_count ++;
            }
            else
            {
                arr[left] = arr[right];
                left ++;
                right --;
                remove_count++;
            }
        }
        else
        {
            left++;
        }
    }
    return len - remove_count;
}

也可以使用快慢指针进行:

如果arr[fast] == target, fast++
如果arr[fast] != target,arr[slow] = arr[fast],slow++

int RemoveElement(int *arr,int len,int target)
{
    int slow = 0;
    for(int i = 0;i<len;++i)
    {
        if(arr[i] != target)
        {
            arr[slow] = arr[i];
            slow++;
        }
    }
    return slow;
}

你可能感兴趣的:(高级数据结构与刷题,算法,数组,移除元素,c/c++,leetcode)