原题链接: 26. 删除有序数组中的重复项
题目大意
给你一个升序排列的数组nums
,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。
由于在某些语言中不能改变数组的长度,所以必须将结果放在数组nums
的第一部分。更规范地说,如果在删除重复项之后有 k
个元素,那么 nums
的前 k
个元素应该保存最终结果。
将最终结果插入 nums
的前 k
个位置后返回 k
。
不要使用额外的空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例:
输入:nums = [1,1,2]
输出:2, nums = [1,2,_]
输入:nums = [0,0,1,1,1,2,2,3,3,4]
输出:5, nums = [0,1,2,3,4]
数据范围:
0 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums 已按 升序 排列
思路
动态维护一个最终返回的数组,长度为k。
遍历原数组,如果当前长度k小于1那肯定遍历的数有效,就加进去(nums[k ++ ] = x)。
长度达到1之后,新的数只要不跟维护的数组的最后一个数相同,则有效,加入,否则,就跳过这个新的数。
最终k就是维护数组的长度,具体见代码。
代码
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int k = 0;
for(auto x : nums){
if(k < 1 || x != nums[k - 1]){
nums[k ++ ] = x;
}
}
return k;
}
};
复杂度分析
时间复杂度:O(n)
空间复杂度:O(1) 原地算法
类似题目
看完本文,你可以解决如下题目:
80. 删除有序数组中的重复项 II
83. 删除排序链表中的重复元素
82. 删除排序链表中的重复元素 II