leetcode27.RemoveElement

Posted 2021-02-05 JESSET

题目描述

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

解法1

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

int removeElement(vector<int>& 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<int>& 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;
}