题目来源
题目描述
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
}
};
题目解析
快排
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int pos = -1;
for (int i = 0; i < nums.size(); ++i) {
if(nums[i] != val){
pos++;
std::swap(nums[i], nums[pos]);
}
}
return pos + 1;
}
};
双指针
根据题意,我们可以将数组分成「前后」两段:
- 前半段是有效部分,存储的是不等于 val 的元素。
- 后半段是无效部分,存储的是等于 val 的元素。
最终答案返回有效部分的结尾下标。
int removeElement(vector<int>& nums, int val) {
int n = nums.size();
int left = 0;
for (int right = 0; right < n; ++right) {
if(nums[right] != val){
nums[left] = val;
left++;
}
}
return left;
}
int removeElement(vector<int>& nums, int val) {
if(nums.size() == 0){
return 0;
}
int low = 0; // 将所有不等于 val的 元素放到low的左边
int high = nums.size() - 1; //将所有val放到high的右边
while (low < high){
// 找到第一个 == val的数
while (low < high && nums[low] != val){
low++;
}
// 找到第一个 != val的数
while(low < high && nums[high] == val){
high--;
}
if(low < high){
std::swap(nums[low], nums[high]);
}
}
// 返回数组的长度
if(nums[low] == val){
return low;
}
// 数组中没有这个val
return nums.size();
}
类似题目
题目 | 思路 |
---|---|
leetcode:26. 删除排序数组中的重复项,每个元素最多不超过1次(原地) Remove Duplicates from Sorted Array | 指针(有效区、基准元素) |
leetcode:80. 删除有序数组中的重复项,每个元素最多不超过2次(原地删除)remove-duplicates-from-sorted-array | 维护两个指针,将数组分成三个区:有效区、无效区间、未知区 |
leetcode:82. 删除有序链表中的重复元素,使得重复元素不再出现 remove-duplicates-from-sorted-list | 递归、双指针 |
leetcode:83. 删除有序链表中的重复元素,使得每个元素只出现一次remove-duplicates-from-sorted-list | 递归、双指针 |
leetcode:27. 删除无序数组中val,使得val出现次数为0(原地,元素顺序可改变)Remove Element | 将数组分成「前后」两段,前半段是有效部分,存储的是不等于 val 的元素,后半段是无效部分,存储的是等于 val 的元素 |
leetcode:203. 删除无序链表中val,使得val出现次数为0(原地,元素顺序可改变)remove-linked-list-elements | 递归:找第一个不需要删除的节点,然后递归 |
leetcode:283. 移动零(原地,元素顺序不可改变)Move Zeroes | 维护两个指针,将数组分成三个区:有效区、无效区间、未知区。有效区,已看不为0的元素;无效区,0;未知区,待看的元素 |