【leetcode】27.RemoveElement

题目描述

给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。
https://leetcode-cn.com/problems/remove-element/

解法1

时间复杂度：O(n)
空间复杂度：O(1)
思路：覆盖法，遍历数组，将非指定值放置在数组前方

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

解法2

时间复杂度：O(n)
空间复杂度：O(1)
思路：第一种方法的改进，当i与j相等时，不需要赋值，避免多余的赋值操作。

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