LeetCode 27 移除元素

LeetCode 27 移除元素

问题：

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

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

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

解法一：

由于题目要求删除数组中等于val 的元素，因此输出数组的长度一定小于等于输入数组的长度，我们可以把输出的数组直接写在输入数组

上。

我们可以遍历整个数组，如果数组中出现等于val的元素，将val元素后面的值，依次前移一个

假设我们要移除的元素为3，如下图所示

• 首先找到需要移除的元素3.如下图①所示
• 接着将3后面的元素，逐次向前覆盖，如下图②所示
• 最终得到结果，如下图③所示

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-oPR3VI5F-1626604078780)(https://i.loli.net/2021/05/30/7NJC6zgUdkr1wZ3.png)]

int removeElement3(int *nums, int numsSize, int val)
{
    //记录数组长度
    int length = numsSize;
    //循环查找数组中值为val的元素
    for (int i = 0; i < numsSize; ++i)
    {
        //如果找到值为val的元素
        if (nums[i] == val)
        {
            //将其后面的元素，逐个向前覆盖
            for (int j = i; j < numsSize - 1; ++j)
            {
                nums[j] = nums[j + 1];
            }
            //找到一个，则数组的个数减一
            length--;
            //由于删除元素的后一个元素放到了刚才移除的位置，这个元素还未判断，所以指针前移一位，判断该元素
            i--;
        }
    }
    return length;
}

这种解法虽然可以解决问题，但其最坏情况下，时间复杂度为O（n^2）,部分测试用例超时。

解法二：双指针法

定义左右两个指针，左指针left指向要被覆盖的位置，右指针right指向要判断的元素

• 如果右指针指向的元素不等于val，说明其为要输出元素，将其复制到左指针所指向位置，左右指针同时右移一位

• 如果右指针指向元素等于val，说明其为要被移除元素，这时左指针不动，右指针右移一位，判断下一位是否为val

  

int removeElement(int *nums, int numsSize, int val)
{
    int left = 0;
    for (int right = 0; right < numsSize; ++right)
    {
        if (nums[right] != val)
        {
            nums[left] = nums[right];
            left++;
        }
    }
    return left;
}

t] = nums[right];
left++;
}
}
return left;
}