LeetCode 之 移除元素
算法模拟: Algorithm Visualizer
在线工具: C++ 在线工具
如果习惯性使用Visual Studio Code进行编译运行,需要C++11特性的支持,可参考博客:
VisualStudio Code 支持C++11插件配置
问题1:LeetCode 27.移除元素
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以 「引用」 方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参作任何拷贝
int len = removeElement(nums, val);// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
for (int i = 0; i < len; i++) {print(nums[i]);
}
示例 1:
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2]
解释:函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为 2 ,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。
示例 2:
输入:nums = [0,1,2,2,3,0,4,2], val = 2
输出:5, nums = [0,1,4,0,3]
解释:函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
C++思路:
可采用双指针方法来修改。
首先,在函数内部,我们使用两个指针 realIndex
和 i
,其中 realIndex
表示当前有效索引, i
用于遍历整个数组。
如果当前遍历到的元素 nums[i]
不等于给定值 val
,则将其赋值给 nums[realIndex]
,然后将 realIndex
增加1,用来指向下一个有效元素。
通过这种方式,我们可以将不等于给定值的元素移动到数组的前面,而等于给定值的元素会被覆盖掉。最后返回 realIndex
即可得到新数组的长度。
class Solution {
public:int removeElement(vector<int>& nums, int val) {// 记录有效索引int realIndex = 0;// 遍历for (int i = 0; i < nums.size(); i++) {// 当数值不同的时候,进行赋值,索引并加1,以此类推,直到结束if (nums[i] != val) {nums[realIndex] = nums[i];realIndex++;}}return realIndex;}
};
时间复杂度: 时间复杂度为O(n),其中n是数组的长度。
问题2: leetCode 26.删除有序数组中重复项
给你一个 升序排列 的数组 nums
,请你** 原地** 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums
中唯一元素的个数。
考虑 nums
的唯一元素的数量为 k
,你需要做以下事情确保你的题解可以被通过:
- 更改数组
nums
,使nums
的前k
个元素包含唯一元素,并按照它们最初在nums
中出现的顺序排列。nums
的其余元素与nums
的大小不重要。 - 返回
k
。
示例 1:
输入:nums = [1,1,2]
输出:2, nums = [1,2,_]
解释:函数应该返回新的长度 2 ,并且原数组 nums 的前两个元素被修改为 1, 2 。不需要考虑数组中超出新长度后面的元素。
示例 2:
输入:nums = [0,0,1,1,1,2,2,3,3,4]
输出:5, nums = [0,1,2,3,4]
解释:函数应该返回新的长度 5 , 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。
C++思路:
class Solution {
public:int removeDuplicates(vector<int>& nums) {int i = 0;for (int j = 1; j < nums.size(); ++j) {if (nums[j] != nums[i]) {i++;nums[i] = nums[j];}}return i + 1;}
};
End