题目:
题目链接
给你一个数组
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;}
};
