leetcode27 移除元素

题目大意

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

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

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

解题思路

设置两个变量left和right,left表示不等于val的区域的下一个位置,right表示当前遍历位置。
(1)nums[right]==val,则right向后移动;
(2)nums[right] != val,则将right位置的元素赋值给left位置,并且left向后移动一位。(0~left-1区域表示不包含val的区域,left位置指的是该区域的下一个位置,因此若遇到一个不是val的元素,将其放到left位置即可,不改变这些元素的相对位置)

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        if (nums.empty())
        	return 0;
        int left = 0, right = 0, length = nums.size();
        for (; right < length; ++right){
        	if (nums[right] != val)
        		nums[left++] = nums[right];
        }
        return left;
    }
};

你可能感兴趣的:(leetcode算法题解)