Leetcode——27 移除元素 双指针的简单应用

Leetcode——27 移除元素 双指针的简单应用

题目描述

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。

假设 nums 中不等于 val 的元素数量为 k，要通过此题，您需要执行以下操作：

• 更改 nums 数组，使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
• 返回 k。

题解

本题需要用两个指针来解决问题，题目中说的是删除所有数值等于val的元素，也可以换个角度想，重构这个序列使里面的数都不为val。

设指针i和k，其中i为正常遍历序列需要的变量，而k是重构序列的长度。

1. 当nums[i] == val：continue 无事发生
2. 当nums[i] != val：nums[k ++] = nums[i] 把这个数存到重构序列当中

重构并不一定意味着需要重新设定数组，由于重构数组的长度一定是动态的小于等于原数组的长度的，即k <= i，因而在同一个数组进行操作也不会导致冲突，最后返回k即可。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        if(nums.size() == 0) return 0;
        int ans = 0;
        int i = 0;
        int k = 0;
        while(i < nums.size()) {
            if(nums[i] != val) nums[k ++] = nums[i];
            i ++;
        }
        return k;
    }
};