【leetcode】 27. 移除元素

2024-08-23 21:44

文章标签 leetcode 元素 27 移除

本文主要是介绍【leetcode】 27. 移除元素，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

题目：

题目链接

给你一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。
假设 nums 中不等于 val 的元素数量为 k，要通过此题，您需要执行以下操作：
更改 nums 数组，使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
返回 k。

代码：

方法一：双指针

class Solution {
public:int removeElement(vector<int>& nums, int val) {int i = 0, j = nums.size() - 1;int cnt = 0; // 统计非 val 元素的个数while (i <= j) { // 注意区间长度为 1 时，也要进入循环while (i <= j && nums[i] != val) { // i 向后移动，找第一个 val++i;cnt++; // 移动过程中统计非 val 的数量}while (i <= j && nums[j] == val) { // j 向前移动，找第一个非 val 元素--j;}if (i < j) { // 交换两个指针所标记的元素int t = nums[i];nums[i] = nums[j];nums[j] = t;}}return cnt;}
};

方法二：平移

class Solution {
public:int removeElement(vector<int>& nums, int val) {int cnt = 0, len = nums.size();for (int i = 0; i < len; i++) {if (nums[i] == val) { // 统计当前位置及其前方 val 的数量 cntcnt++;} else{nums[i - cnt] = nums[i]; // 将不是 val 的值向前移动 cnt 个位置}}return nums.size() - cnt;}
};